diff --git a/upgrade.php b/upgrade.php
index c1010dad48..9847387b68 100644
--- a/upgrade.php
+++ b/upgrade.php
@@ -9,7 +9,7 @@ if (($username=='root') || ($username=='admin')) {
 }
 
 
-($argv[0]) ? $branch = $argv[0] : $branch = 'master';
+($argv[1]) ? $branch = $argv[1] : $branch = 'master';
 
 echo "Welcome to the Snipe-IT upgrader.\n\n";
 echo "Please note that this script will not download the latest Snipe-IT \n";