diff --git a/modules/nixos/graphical/rofi/rofi-prompt.sh b/modules/nixos/graphical/rofi/rofi-prompt.sh
index 85ca923..5c87ddd 100755
--- a/modules/nixos/graphical/rofi/rofi-prompt.sh
+++ b/modules/nixos/graphical/rofi/rofi-prompt.sh
@@ -42,6 +42,6 @@ chosen=$(printf '%s;%s\n' "$yes" "$no" |
         -selected-row 1)
 
 case "$chosen" in
-    "$yes") exit 0 ;;
-    *) exit 1 ;;
+    "$yes") return 0 ;;
+    *) return 1 ;;
 esac