diff --git a/dist/platforms/ubuntu/run_tests.sh b/dist/platforms/ubuntu/run_tests.sh
index e80e13cc..a4b6c574 100755
--- a/dist/platforms/ubuntu/run_tests.sh
+++ b/dist/platforms/ubuntu/run_tests.sh
@@ -36,6 +36,25 @@ echo "Using custom parameters $CUSTOM_PARAMETERS."
 
 echo "Using Unity version \"$UNITY_VERSION\" to test."
 
+#
+# Setup token for private package registry.
+#
+
+if [ -n "$PRIVATE_REGISTRY_TOKEN" ]; then
+  echo "Private registry token detected, creating .upmconfig.toml"
+
+  UPM_CONFIG_TOML_PATH="$HOME/.upmconfig.toml"
+  echo "Creating toml at path: $UPM_CONFIG_TOML_PATH"
+
+  touch $UPM_CONFIG_TOML_PATH
+
+  cat > "$UPM_CONFIG_TOML_PATH" <<EOF
+  [npmAuth."$SCOPED_REGISTRY_URL"]
+  token = "$PRIVATE_REGISTRY_TOKEN"
+  alwaysAuth = true
+EOF
+fi
+
 #
 # Create an empty project for testing if in package mode
 #
@@ -120,20 +139,6 @@ if [ "$PACKAGE_MODE" = "true" ]; then
 
   UNITY_PROJECT_PATH="$TEMP_PROJECT_PATH"
 
-  if [ -n "$PRIVATE_REGISTRY_TOKEN" ]; then
-    echo "Private registry token detected, creating .upmconfig.toml"
-
-    UPM_CONFIG_TOML_PATH="$HOME/.upmconfig.toml"
-    echo "Creating toml at path: $UPM_CONFIG_TOML_PATH"
-
-    touch $UPM_CONFIG_TOML_PATH
-
-    cat > "$UPM_CONFIG_TOML_PATH" <<EOF
-    [npmAuth."$SCOPED_REGISTRY_URL"]
-    token = "$PRIVATE_REGISTRY_TOKEN"
-    alwaysAuth = true
-EOF
-  fi
 
 fi
 
@@ -280,4 +285,4 @@ if [ -d "$FULL_COVERAGE_RESULTS_PATH" ]; then
   chmod -R a+r "$FULL_COVERAGE_RESULTS_PATH"
 else
   echo "Coverage results directory does not exist. If you are expecting coverage results, please make sure the Code Coverage package is installed in your unity project and that it is set up correctly."
-fi
\ No newline at end of file
+fi