Update the test to do something after handshake