diff --git a/tests/_graphical_wait_login.pm b/tests/_graphical_wait_login.pm
index df9c6fc9..2123a29a 100644
--- a/tests/_graphical_wait_login.pm
+++ b/tests/_graphical_wait_login.pm
@@ -26,7 +26,7 @@ sub run {
         $wait_time = 300;
     }
     # Wait for the login screen unless it is not already booted
-    unless (get_var("ASSUME_BOOT")) { 
+    unless (get_var("ASSUME_BOOT") == 1) { 
         boot_to_login_screen(timeout => $wait_time);
     }
     # GDM 3.24.1 dumps a cursor in the middle of the screen here...
diff --git a/tests/upgrade_preinstall.pm b/tests/upgrade_preinstall.pm
index e5bcf2ee..ebaf12ea 100644
--- a/tests/upgrade_preinstall.pm
+++ b/tests/upgrade_preinstall.pm
@@ -26,7 +26,7 @@ sub run {
     assert_script_run $update_command, 600;
 }
 
-if (get_var("ASSUME_BOOT")) {
+if (get_var("ASSUME_BOOT") == 1) {
     desktop_vt;
 }