diff --git a/session-manager.lua b/session-manager.lua
index ef08a5b..d7ae567 100644
--- a/session-manager.lua
+++ b/session-manager.lua
@@ -102,7 +102,13 @@ local function recreate_workspace(window, workspace_data)
   local foreground_process = initial_pane:get_foreground_process_name()
 
   -- Check if the foreground process is a shell
-  if foreground_process:find("sh") or foreground_process:find("cmd.exe") or foreground_process:find("powershell.exe") or foreground_process:find("pwsh.exe") or foreground_process:find("nu") then
+  if foreground_process and (
+        foreground_process:find("sh")
+        or foreground_process:find("cmd.exe")
+        or foreground_process:find("powershell.exe")
+        or foreground_process:find("pwsh.exe")
+        or foreground_process:find("nu")
+    ) then
     -- Safe to close
     initial_pane:send_text("exit\r")
   else