diff --git a/transport/mock/mock.go b/transport/mock/mock.go
index 5c0d3258..d544d96b 100644
--- a/transport/mock/mock.go
+++ b/transport/mock/mock.go
@@ -63,7 +63,12 @@ func (ms *mockSocket) Send(m *transport.Message) error {
 }
 
 func (ms *mockSocket) Close() error {
-	close(ms.exit)
+	select {
+	case <-ms.exit:
+		return nil
+	default:
+		close(ms.exit)
+	}
 	return nil
 }
 
@@ -72,7 +77,12 @@ func (m *mockListener) Addr() string {
 }
 
 func (m *mockListener) Close() error {
-	close(m.exit)
+	select {
+	case <-m.exit:
+		return nil
+	default:
+		close(m.exit)
+	}
 	return nil
 }