diff --git a/nixos-edit.script.nix b/nixos-edit.script.nix
index b46dd11..44d3100 100644
--- a/nixos-edit.script.nix
+++ b/nixos-edit.script.nix
@@ -24,7 +24,7 @@ pkgs.writeShellScriptBin "nixos-edit" ''
 		echo dir setup done
 		echo "> setting up ssh"
 		echo ssh setup
-		if ! [ -e ~/.ssh/id_* ] ; then
+		if ! [ -e ~/.ssh/id_*.pub ] ; then
 			ssh-keygen || exit
 		fi
 		echo ssh key created