diff --git a/src/main/java/com/browserstack/gradle/BrowserStackTask.java b/src/main/java/com/browserstack/gradle/BrowserStackTask.java
index d8f332c..929577c 100644
--- a/src/main/java/com/browserstack/gradle/BrowserStackTask.java
+++ b/src/main/java/com/browserstack/gradle/BrowserStackTask.java
@@ -34,6 +34,7 @@ public class BrowserStackTask extends DefaultTask {
 
   private String appVariantBaseName = "debug";
 
+  @Input
   public String command ;
 
   public void setAppVariantBaseName(String appVariantBaseName) {