1 files changed, 4 insertions, 2 deletions
diff --git a/src/server.c b/src/server.c
index b189d56..faa4c4c 100644
--- a/src/server.c
+++ b/src/server.c
@@ -176,14 +176,16 @@ static int dh_handshake_server(ssh_session session) {
     return -1;
   }
 
-  switch(session->hostkeys){
+  switch(session->srv.hostkey) {
     case SSH_KEYTYPE_DSS:
       privkey = session->srv.dsa_key;
       break;
     case SSH_KEYTYPE_RSA:
+    case SSH_KEYTYPE_RSA1:
       privkey = session->srv.rsa_key;
       break;
-    default:
+    case SSH_KEYTYPE_ECDSA:
+    case SSH_KEYTYPE_UNKNOWN:
       privkey = NULL;
   }