diff --git a/app/Actions/Server/UpdateCoolify.php b/app/Actions/Server/UpdateCoolify.php
index 6803b0d9e..d57a4fe46 100644
--- a/app/Actions/Server/UpdateCoolify.php
+++ b/app/Actions/Server/UpdateCoolify.php
@@ -56,7 +56,7 @@ class UpdateCoolify
 
         remote_process([
             'curl -fsSL https://cdn.coollabs.io/coolify/upgrade.sh -o /data/coolify/source/upgrade.sh',
-            "bash /data/coolify/source/upgrade.sh --latest-coolify-version $this->latestVersion",
+            "bash /data/coolify/source/upgrade.sh $this->latestVersion",
         ], $this->server);
     }
 }
diff --git a/other/nightly/upgrade.sh b/other/nightly/upgrade.sh
index 438ca50ee..670072b12 100644
--- a/other/nightly/upgrade.sh
+++ b/other/nightly/upgrade.sh
@@ -3,31 +3,12 @@
 
 VERSION="13"
 CDN="https://cdn.coollabs.io/coolify-nightly"
+LATEST_IMAGE=${1:-latest}
+LATEST_HELPER_VERSION=${2:-latest}
+
 DATE=$(date +%Y-%m-%d-%H-%M-%S)
 LOGFILE="/data/coolify/source/upgrade-${DATE}.log"
 
-# Parse named arguments
-while [ $# -gt 0 ]; do
-    case "$1" in
-        --latest-coolify-version)
-            LATEST_IMAGE="$2"
-            shift 2
-            ;;
-        --latest-helper-version)
-            LATEST_HELPER_VERSION="$2"
-            shift 2
-            ;;
-        *)
-            echo "Unknown parameter: $1"
-            exit 1
-            ;;
-    esac
-done
-
-# Set defaults if not provided
-LATEST_IMAGE=${LATEST_IMAGE:-latest}
-LATEST_HELPER_VERSION=${LATEST_HELPER_VERSION:-latest}
-
 curl -fsSL $CDN/docker-compose.yml -o /data/coolify/source/docker-compose.yml
 curl -fsSL $CDN/docker-compose.prod.yml -o /data/coolify/source/docker-compose.prod.yml
 curl -fsSL $CDN/.env.production -o /data/coolify/source/.env.production
diff --git a/scripts/upgrade.sh b/scripts/upgrade.sh
index ac9387c8c..1bdf0f461 100644
--- a/scripts/upgrade.sh
+++ b/scripts/upgrade.sh
@@ -3,31 +3,12 @@
 
 VERSION="13"
 CDN="https://cdn.coollabs.io/coolify"
+LATEST_IMAGE=${1:-latest}
+LATEST_HELPER_VERSION=${2:-latest}
+
 DATE=$(date +%Y-%m-%d-%H-%M-%S)
 LOGFILE="/data/coolify/source/upgrade-${DATE}.log"
 
-# Parse named arguments
-while [ $# -gt 0 ]; do
-    case "$1" in
-        --latest-coolify-version)
-            LATEST_IMAGE="$2"
-            shift 2
-            ;;
-        --latest-helper-version)
-            LATEST_HELPER_VERSION="$2"
-            shift 2
-            ;;
-        *)
-            echo "Unknown parameter: $1"
-            exit 1
-            ;;
-    esac
-done
-
-# Set defaults if not provided
-LATEST_IMAGE=${LATEST_IMAGE:-latest}
-LATEST_HELPER_VERSION=${LATEST_HELPER_VERSION:-latest}
-
 curl -fsSL $CDN/docker-compose.yml -o /data/coolify/source/docker-compose.yml
 curl -fsSL $CDN/docker-compose.prod.yml -o /data/coolify/source/docker-compose.prod.yml
 curl -fsSL $CDN/.env.production -o /data/coolify/source/.env.production