diff --git a/ebssurrogate/scripts/nix-provision.sh b/ebssurrogate/scripts/nix-provision.sh
index 41a81ea11..4063ddca9 100755
--- a/ebssurrogate/scripts/nix-provision.sh
+++ b/ebssurrogate/scripts/nix-provision.sh
@@ -19,7 +19,7 @@ function cleanup_apt {
 
 function update_and_upgrade_apt {
 	apt-get update --yes
-	apt-get upgrade --yes
+	apt-get dist-upgrade --yes
 }
 
 function install_packages {
diff --git a/ebssurrogate/scripts/qemu-bootstrap-nix.sh b/ebssurrogate/scripts/qemu-bootstrap-nix.sh
index 6a9d4df1f..c064454b0 100755
--- a/ebssurrogate/scripts/qemu-bootstrap-nix.sh
+++ b/ebssurrogate/scripts/qemu-bootstrap-nix.sh
@@ -50,7 +50,7 @@ function cleanup_apt {
 
 function update_and_upgrade_apt {
 	apt-get update --yes
-	apt-get upgrade --yes
+	apt-get dist-upgrade --yes
 }
 
 function waitfor_boot_finished {
diff --git a/ebssurrogate/scripts/surrogate-bootstrap-nix.sh b/ebssurrogate/scripts/surrogate-bootstrap-nix.sh
index 81fae3100..2f43a5a45 100755
--- a/ebssurrogate/scripts/surrogate-bootstrap-nix.sh
+++ b/ebssurrogate/scripts/surrogate-bootstrap-nix.sh
@@ -48,7 +48,7 @@ function setup_apt {
 
 function update_and_upgrade_apt {
 	apt-get update --yes
-	apt-get upgrade --yes
+	apt-get dist-upgrade --yes
 }
 
 function waitfor_boot_finished {