diff --git a/cd/__init__.py b/cd/__init__.py
index 28b8c3b..05ae1d1 100644
--- a/cd/__init__.py
+++ b/cd/__init__.py
@@ -87,8 +87,7 @@ def install_commands(path):
 
 
 def set_ssh(*arguments):
-    ssh_hosts = arguments[0]
-    ssh_key = projects.environment.get('CD', None)
+    ssh_key, ssh_hosts = arguments
     ssh_type = projects.environment.get('CD_SSH_TYPE', 'ed25519')
     #
     home = os.path.expanduser('~')