diff --git a/build/image-optimization.sh b/build/image-optimization.sh
index 81866a0fee..6d494b8ddb 100755
--- a/build/image-optimization.sh
+++ b/build/image-optimization.sh
@@ -1,7 +1,7 @@
 #!/usr/bin/env bash
 
 CHECK_DIR='../'
-if ! [ "$SERVER_VERSION" ]; then
+if [[ -d "$1" ]]; then
 	CHECK_DIR=$1
 fi