1 files changed, 10 insertions, 3 deletions
diff --git a/profiles/i3.py b/profiles/i3.py
index 3283848e..37029a02 100644
--- a/profiles/i3.py
+++ b/profiles/i3.py
@@ -1,6 +1,8 @@
 # Common package for i3, lets user select which i3 configuration they want.
 
 import archinstall
+from archinstall import Menu
+from archinstall.lib.menu.menu import MenuSelectionType
 
 is_top_level_profile = False
 
@@ -27,13 +29,16 @@ def _prep_function(*args, **kwargs):
 
 	supported_configurations = ['i3-wm', 'i3-gaps']
 
-	desktop = archinstall.Menu('Select your desired configuration', supported_configurations).run()
+	choice = Menu('Select your desired configuration', supported_configurations).run()
 
-	if desktop:
+	if choice.type_ != MenuSelectionType.Selection:
+		return False
+
+	if choice.value:
 		# Temporarily store the selected desktop profile
 		# in a session-safe location, since this module will get reloaded
 		# the next time it gets executed.
-		archinstall.storage['_i3_configuration'] = desktop
+		archinstall.storage['_i3_configuration'] = choice.value
 
 		# i3 requires a functioning Xorg installation.
 		profile = archinstall.Profile(None, 'xorg')
@@ -43,6 +48,8 @@ def _prep_function(*args, **kwargs):
 			else:
 				print('Deprecated (??): xorg profile has no _prep_function() anymore')
 
+	return False
+
 
 if __name__ == 'i3':
 	"""