diff --git a/resources/bin/bal b/resources/bin/bal
index 21dcd7f6..da1531f1 100755
--- a/resources/bin/bal
+++ b/resources/bin/bal
@@ -66,7 +66,7 @@
   RUN_COMMAND=false
   RUN_BALLERINA=true
 
-  if [ "$1" == "dist" ] || [ "$2" == "dist" ] || [ "$1" == "update" ]|| [ "$2" == "update" ]
+  if [ "$1" == "dist" ] || [ "$2" == "dist" ] || [ "$1" == "update" ] || ( [ "$1" == "dist" ] && [ "$2" == "update" ] )
   then
       RUN_COMMAND=true
       RUN_BALLERINA=false
diff --git a/resources/bin/bal.bat b/resources/bin/bal.bat
index 5ed4f353..3882d2f8 100644
--- a/resources/bin/bal.bat
+++ b/resources/bin/bal.bat
@@ -29,7 +29,11 @@ set JAVA_CMD=java
 if "%1" == "dist" set dist=true
 if "%~2" == "dist" set dist=true
 if "%1" == "update" set dist=true
-if "%~2" == "update" set dist=true
+if "%~2" == "update" (
+    if "%1" == "dist" (
+        set dist=true
+    )
+)
 if "%1" == "build" set dist=true
 if "%1" == "update" set update=true
 if "%1" == "build" set build=true