diff --git a/session_test.go b/session_test.go index 179204e86..56e1c03ec 100644 --- a/session_test.go +++ b/session_test.go @@ -1356,7 +1356,7 @@ var _ = Describe("Session", func() { done := make(chan struct{}) go func() { defer GinkgoRecover() - cryptoSetup.EXPECT().RunHandshake() + cryptoSetup.EXPECT().RunHandshake().MaxTimes(1) close(sess.handshakeCompleteChan) err := sess.run() nerr, ok := err.(net.Error)