diff --git a/app/Http/Livewire/CheckUpdate.php b/app/Http/Livewire/CheckUpdate.php
index 069cf60c9..3741f9475 100644
--- a/app/Http/Livewire/CheckUpdate.php
+++ b/app/Http/Livewire/CheckUpdate.php
@@ -15,12 +15,11 @@ class CheckUpdate extends Component
 
     protected function upgrade()
     {
-        $server = Server::where('ip', 'host.docker.internal')->first();
+        $server = Server::where('ip', 'coolify-testing-host')->first();
         if (!$server) {
             return;
         }
-        runRemoteCommandSync($server, ['curl -fsSL https://raw.githubusercontent.com/coollabsio/coolify/v4/scripts/upgrade.sh -o /data/coolify/source/upgrade.sh']);
-        runRemoteCommandSync($server, ["bash /data/coolify/source/upgrade.sh $this->latestVersion"]);
+        runRemoteCommandSync($server, ["curl -fsSL https://raw.githubusercontent.com/coollabsio/coolify/v4/scripts/upgrade.sh -o /root/upgrade.sh && bash -x /root/upgrade.sh $this->latestVersion"]);
     }
     public function forceUpgrade()
     {
diff --git a/bootstrap/helpers.php b/bootstrap/helpers.php
index f9be10786..46028e545 100644
--- a/bootstrap/helpers.php
+++ b/bootstrap/helpers.php
@@ -128,7 +128,7 @@ if (!function_exists('runRemoteCommandSync')) {
         $exitCode = $process->exitCode();
         if ($exitCode !== 0) {
             Log::error($output);
-            // throw new \RuntimeException('There was an error running the command.');
+            throw new \RuntimeException('There was an error running the command.');
         }
         return $output;
     }
diff --git a/resources/views/livewire/check-update.blade.php b/resources/views/livewire/check-update.blade.php
index dcd2f64b6..22cd68a44 100644
--- a/resources/views/livewire/check-update.blade.php
+++ b/resources/views/livewire/check-update.blade.php
@@ -1,7 +1,7 @@
 <div>
     <button wire:click='checkUpdate'>Updates</button>
-    @env('production')
-    <button wire:click='forceUpgrade'>Force Upgrade</button>
-    @endenv
+    @if (auth()->user()->teams->contains(0))
+        <button wire:click='forceUpgrade'>Force Upgrade</button>
+    @endif
     {{ $updateAvailable ? 'Update available' : 'No updates' }}
 </div>
diff --git a/scripts/upgrade.sh b/scripts/upgrade.sh
index 3fb777943..f7e52bdd1 100644
--- a/scripts/upgrade.sh
+++ b/scripts/upgrade.sh
@@ -1,4 +1,4 @@
 #!/bin/bash
 ## Do not modify this file. You will lost the ability to autoupdate!
 
-docker run --pull always -v /data/coolify/source:/data/coolify/source -v /var/run/docker.sock:/var/run/docker.sock -ti --rm ghcr.io/coollabsio/coolify-builder bash -c "APP_TAG=${1:-} docker compose --env-file /data/coolify/source/.env -f /data/coolify/source/docker-compose.yml -f /data/coolify/source/docker-compose.prod.yml up -d --pull always --remove-orphans --force-recreate"
+docker run --pull always -v /data/coolify/source:/data/coolify/source -v /var/run/docker.sock:/var/run/docker.sock --rm ghcr.io/coollabsio/coolify-builder bash -c "APP_TAG=${1:-} docker compose --env-file /data/coolify/source/.env -f /data/coolify/source/docker-compose.yml -f /data/coolify/source/docker-compose.prod.yml up -d --pull always --remove-orphans --force-recreate"