Fix 1117 (#1126)

* Fix 1117

* Update

* flake8

Co-authored-by: Daniel Girtler <girtler.daniel@gmail.com>

1 files changed, 3 insertions, 2 deletions

diff --git a/profiles/sway.py b/profiles/sway.py
index e9c71b79..0819db95 100644
--- a/profiles/sway.py
+++ b/profiles/sway.py
@@ -1,5 +1,6 @@
 # A desktop environment using "Sway"
 import archinstall
+from archinstall import Menu
 
 is_top_level_profile = False
 
@@ -22,8 +23,8 @@ def _check_driver() -> bool:
 
 	if packages and "nvidia" in packages:
 		prompt = 'The proprietary Nvidia driver is not supported by Sway. It is likely that you will run into issues, are you okay with that?'
-		choice = archinstall.Menu(prompt, ['yes', 'no'], default_option='no').run()
-		if choice == 'no':
+		choice = Menu(prompt, Menu.yes_no(), default_option=Menu.no()).run()
+		if choice == Menu.no():
 			return False
 
 	return True