2 files changed, 5 insertions, 5 deletions

diff --git a/kex.c b/kex.c
index f4c130f14..8800d4001 100644
--- a/kex.c
+++ b/kex.c
@@ -1,4 +1,4 @@
-/* $OpenBSD: kex.c,v 1.121 2016/09/12 23:31:27 djm Exp $ */
+/* $OpenBSD: kex.c,v 1.122 2016/09/19 19:02:19 markus Exp $ */
 /*
  * Copyright (c) 2000, 2001 Markus Friedl.  All rights reserved.
  *
@@ -425,6 +425,8 @@ kex_input_newkeys(int type, u_int32_t seq, void *ctxt)
         ssh_dispatch_set(ssh, SSH2_MSG_NEWKEYS, &kex_protocol_error);
         if ((r = sshpkt_get_end(ssh)) != 0)
                 return r;
+        if ((r = ssh_set_newkeys(ssh, MODE_IN)) != 0)
+                return r;
         kex->done = 1;
         sshbuf_reset(kex->peer);
         /* sshbuf_reset(kex->my); */
diff --git a/packet.c b/packet.c
index 711091da7..fb316acbc 100644
--- a/packet.c
+++ b/packet.c
@@ -1,4 +1,4 @@
-/* $OpenBSD: packet.c,v 1.237 2016/09/12 01:22:38 deraadt Exp $ */
+/* $OpenBSD: packet.c,v 1.238 2016/09/19 19:02:19 markus Exp $ */
 /*
  * Author: Tatu Ylonen <ylo@cs.hut.fi>
  * Copyright (c) 1995 Tatu Ylonen <ylo@cs.hut.fi>, Espoo, Finland
@@ -1907,9 +1907,7 @@ ssh_packet_read_poll2(struct ssh *ssh, u_char *typep, u_int32_t *seqnr_p)
                         return r;
                 return SSH_ERR_PROTOCOL_ERROR;
         }
-        if (*typep == SSH2_MSG_NEWKEYS)
-                r = ssh_set_newkeys(ssh, MODE_IN);
-        else if (*typep == SSH2_MSG_USERAUTH_SUCCESS && !state->server_side)
+        if (*typep == SSH2_MSG_USERAUTH_SUCCESS && !state->server_side)
                 r = ssh_packet_enable_delayed_compress(ssh);
         else
                 r = 0;