diff --git a/modules/users.nix b/modules/users.nix
index a1607c5..5192f87 100644
--- a/modules/users.nix
+++ b/modules/users.nix
@@ -48,7 +48,7 @@
 
   services.mingetty.autologinUser = "balsoft";
 
-  environment.loginShellInit = "sway; sudo -u balsoft zsh";
+  environment.loginShellInit = "set -e; [[ $USER == balsoft ]] && (sway; sudo bash || exit 1)";
 
   security.pam.services = builtins.listToAttrs (builtins.map (name: {
     inherit name;