diff --git a/stdlib/sql_test.go b/stdlib/sql_test.go
index bf99a8bb..65f80ac4 100644
--- a/stdlib/sql_test.go
+++ b/stdlib/sql_test.go
@@ -1211,6 +1211,7 @@ func TestStmtQueryContextCancel(t *testing.T) {
 		pgmock.ExpectMessage(&pgproto3.Sync{}),
 
 		pgmock.SendMessage(&pgproto3.BindComplete{}),
+		pgmock.WaitForClose(),
 	)
 
 	server, err := pgmock.NewServer(script)