Fix handshake bugs