diff --git a/net/mock/mock_test.go b/net/mock/mock_test.go
index e19d95ef4..a49e205e7 100644
--- a/net/mock/mock_test.go
+++ b/net/mock/mock_test.go
@@ -370,7 +370,7 @@ func TestStreamsStress(t *testing.T) {
 		}(i)
 	}
 
-	wg.Done()
+	wg.Wait()
 }
 
 func TestAdding(t *testing.T) {