diff --git a/tests/asyncio/test_retry_async.py b/tests/asyncio/test_retry_async.py
index b3ee681e..53b04799 100644
--- a/tests/asyncio/test_retry_async.py
+++ b/tests/asyncio/test_retry_async.py
@@ -138,9 +138,8 @@ async def test_retry_target_bad_sleep_generator():
 async def test_retry_streaming_target_bad_sleep_generator():
     from google.api_core.retry_streaming_async import retry_target_stream
 
-    gen = retry_target_stream(None, lambda x: True, [], None)
     with pytest.raises(ValueError, match="Sleep generator"):
-        await gen.__anext__()
+        await retry_target_stream(None, None, [], None).__anext__()
 
 
 class TestAsyncRetry:
diff --git a/tests/unit/test_retry.py b/tests/unit/test_retry.py
index 3e0ecd51..61868e1b 100644
--- a/tests/unit/test_retry.py
+++ b/tests/unit/test_retry.py
@@ -166,11 +166,10 @@ def test_retry_target_bad_sleep_generator():
 def test_retry_streaming_target_bad_sleep_generator():
     from google.api_core.retry_streaming import retry_target_stream
 
-    gen = retry_target_stream(None, lambda x: True, [], None)
     with pytest.raises(
         ValueError, match="Sleep generator stopped yielding sleep values"
     ):
-        next(gen)
+        next(retry_target_stream(None, None, [], None))
 
 
 class TestRetry(object):