diff --git a/code-server/run.sh b/code-server/run.sh
index 55c370c..26a1c6e 100755
--- a/code-server/run.sh
+++ b/code-server/run.sh
@@ -71,16 +71,16 @@ for extension in "$${EXTENSIONLIST[@]}"; do
 done
 
 if [ "${AUTO_INSTALL_EXTENSIONS}" = true ]; then
-  WORKSPACE_DIR="$HOME/"
-  if [ -n "${FOLDER}" ]; then
-    WORKSPACE_DIR="${FOLDER}"
-  fi
-
   if ! command -v jq > /dev/null; then
     echo "jq is required to install extensions from a workspace file."
     exit 0
   fi
 
+  WORKSPACE_DIR="$HOME"
+  if [ -n "${FOLDER}" ]; then
+    WORKSPACE_DIR="${FOLDER}"
+  fi
+
   if [ -f "$WORKSPACE_DIR/.vscode/extensions.json" ]; then
     printf "🧩 Installing extensions from %s/.vscode/extensions.json...\n" "$WORKSPACE_DIR"
     extensions=$(jq -r '.recommendations[]' "$WORKSPACE_DIR"/.vscode/extensions.json)