diff --git a/wsnet/listen.go b/wsnet/listen.go
index 859ba17f..c11df79c 100644
--- a/wsnet/listen.go
+++ b/wsnet/listen.go
@@ -49,7 +49,7 @@ func Listen(ctx context.Context, broker string) (io.Closer, error) {
 					case <-ctx.Done():
 						err = ctx.Err()
 					}
-					if errors.Is(err, context.Canceled) || errors.Is(err, context.DeadlineExceeded) {
+					if err == nil || errors.Is(err, context.Canceled) || errors.Is(err, context.DeadlineExceeded) {
 						break
 					}
 				}