fixed the wrong check (due to delayed expansion) in bat-template

sbt · Dec 15, 2014 · 16b8c52 · 16b8c52
1 parent bfc7158
commit 16b8c52
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/resources/com/typesafe/sbt/packager/archetypes/bat-template b/src/main/resources/com/typesafe/sbt/packager/archetypes/bat-template
@@ -114,8 +114,8 @@ if "!_TEST_PARAM:~0,2!"=="-D" (
     )
   )
 ) else (
-  if "%1"=="-main" (
-    set CUSTOM_MAIN_CLASS=%2
+  if "!_TEST_PARAM!"=="-main" (
+    call set CUSTOM_MAIN_CLASS=%%2
     shift
   ) else (
     set _APP_ARGS=!_APP_ARGS! !_PARAM1!