diff --git a/files/p1/boot/grub/grub.cfg b/files/p1/boot/grub/grub.cfg
index c772f698..7b1d5b46 100644
--- a/files/p1/boot/grub/grub.cfg
+++ b/files/p1/boot/grub/grub.cfg
@@ -196,17 +196,16 @@ menuentry 'Poweroff' ${menuentry_id_option} poweroff {
   halt
 }
 
-if [ ${vesa_mode} = 1 ]; then
-  menuentry 'Change Vesa to Text Output' ${menuentry_id_option} videomode {
-    set vesa_mode=0
-    save_env vesa_mode
+if [ "${linux_gfx_mode}" = "keep" ]; then
+  menuentry 'Change vesa to text video mode' ${menuentry_id_option} videomode {
+    set linux_gfx_mode=text
+    save_env linux_gfx_mode
     configfile ${prefix}/grub.cfg
   }
 else
-  menuentry 'Change Text to Vesa Output' ${menuentry_id_option} videomode {
-    set vesa_mode=1
-    save_env vesa_mode
-    reboot
+  menuentry 'Change text to vesa video mode' ${menuentry_id_option} videomode {
+    set linux_gfx_mode=keep
+    save_env linux_gfx_mode
     configfile ${prefix}/grub.cfg
   }
 fi
\ No newline at end of file