diff --git a/build.gradle b/build.gradle
index 26a359d..c0e8e25 100644
--- a/build.gradle
+++ b/build.gradle
@@ -4,7 +4,7 @@ plugins {
     id 'com.gradle.plugin-publish' version '0.11.0'
 }
 
-version '3.1.0'
+version '3.1.1'
 
 pluginBundle {
     website = 'https://www.browserstack.com'
@@ -17,7 +17,7 @@ pluginBundle {
             description = 'Runs Espresso tests on BrowserStack'
             tags = ['espresso', 'test', 'browserstack', 'app', 'automate',
                 'app-automate', 'appautomate', 'app-live', 'applive']
-            version = '3.1.0'
+            version = '3.1.1'
         }
     }
 }
diff --git a/src/main/java/com/browserstack/gradle/BrowserStackTask.java b/src/main/java/com/browserstack/gradle/BrowserStackTask.java
index 929577c..02c77b5 100644
--- a/src/main/java/com/browserstack/gradle/BrowserStackTask.java
+++ b/src/main/java/com/browserstack/gradle/BrowserStackTask.java
@@ -17,6 +17,8 @@
 import org.gradle.api.DefaultTask;
 import org.gradle.api.tasks.Input;
 import org.jetbrains.annotations.NotNull;
+import org.gradle.api.tasks.Optional;
+
 
 public class BrowserStackTask extends DefaultTask {
 
@@ -35,6 +37,7 @@ public class BrowserStackTask extends DefaultTask {
   private String appVariantBaseName = "debug";
 
   @Input
+  @Optional
   public String command ;
 
   public void setAppVariantBaseName(String appVariantBaseName) {