diff --git a/connect_ext_test.go b/connect_ext_test.go
index 5e3f775b..5e82bfce 100644
--- a/connect_ext_test.go
+++ b/connect_ext_test.go
@@ -2914,7 +2914,11 @@ func (p pingServer) CountUp(
 		if err := stream.Send(&pingv1.CountUpResponse{Number: i}); err != nil {
 			return err
 		}
-		time.Sleep(p.delayCountUp)
+		select {
+		case <-ctx.Done():
+			return ctx.Err()
+		case <-time.After(p.delayCountUp):
+		}
 	}
 	return nil
 }