diff --git a/scripts/restore.sh b/scripts/restore.sh
index 1a5e3f98..3645485c 100755
--- a/scripts/restore.sh
+++ b/scripts/restore.sh
@@ -109,8 +109,9 @@ tmux_socket() {
 cache_tmux_default_command() {
 	local default_shell="$(get_tmux_option "default-shell" "")"
 	local opt=""
-	if [ "$(basename "$default_shell")" == "bash" ]; then
-		opt="-l "
+	local available_shells=(zsh bash)
+  if [[ " ${available_shells[*]} " =~ " $(basename "$default_shell") " ]]; then
+    opt="-l "
 	fi
 	export TMUX_DEFAULT_COMMAND="$(get_tmux_option "default-command" "$opt$default_shell")"
 }