diff --git a/pkg/cli/cli.go b/pkg/cli/cli.go
index 5caa6f7..7736ab4 100644
--- a/pkg/cli/cli.go
+++ b/pkg/cli/cli.go
@@ -30,7 +30,7 @@ func Build() *cobra.Command {
 			}
 			defer client.CleanUp()
 
-			return tasks.Build(client, imageTag)
+			return tasks.Build(client, imageTag, "main.go")
 		},
 	}