diff --git a/sdk/eventhub/azure-eventhub/tests/livetest/asynctests/test_consumer_client_async.py b/sdk/eventhub/azure-eventhub/tests/livetest/asynctests/test_consumer_client_async.py
index 41c2a950b629..0ff6ae711cbf 100644
--- a/sdk/eventhub/azure-eventhub/tests/livetest/asynctests/test_consumer_client_async.py
+++ b/sdk/eventhub/azure-eventhub/tests/livetest/asynctests/test_consumer_client_async.py
@@ -16,7 +16,7 @@ async def test_receive_no_partition_async(connstr_senders):
 
     async def on_event(partition_context, event):
         on_event.received += 1
-        await partition_context.update_checkpoint(event, fake_kwarg="arg")  # ignores fake_kwarg
+        await partition_context.update_checkpoint(event)
         on_event.namespace = partition_context.fully_qualified_namespace
         on_event.eventhub_name = partition_context.eventhub_name
         on_event.consumer_group = partition_context.consumer_group
diff --git a/sdk/eventhub/azure-eventhub/tests/livetest/asynctests/test_receive_async.py b/sdk/eventhub/azure-eventhub/tests/livetest/asynctests/test_receive_async.py
index cd7de4f46037..6fb153052023 100644
--- a/sdk/eventhub/azure-eventhub/tests/livetest/asynctests/test_receive_async.py
+++ b/sdk/eventhub/azure-eventhub/tests/livetest/asynctests/test_receive_async.py
@@ -10,7 +10,7 @@
 
 from azure.eventhub import EventData, TransportType
 from azure.eventhub.exceptions import EventHubError
-from azure.eventhub.aio import EventHubConsumerClient
+from azure.eventhub.aio import EventHubProducerClient, EventHubConsumerClient
 
 
 @pytest.mark.liveTest
@@ -30,16 +30,27 @@ async def on_event(partition_context, event):
 
     on_event.called = False
     connection_str, senders = connstr_senders
+    # test async producer client
+    producer_client = EventHubProducerClient.from_connection_string(connection_str)
+    partitions = await producer_client.get_partition_ids()
+    senders = []
+    for p in partitions:
+        sender = producer_client._create_producer(partition_id=p)
+        senders.append(sender)
+    
     client = EventHubConsumerClient.from_connection_string(connection_str, consumer_group='$default')
     async with client:
         task = asyncio.ensure_future(client.receive(on_event, partition_id="0", starting_position="@latest"))
         await asyncio.sleep(10)
         assert on_event.called is False
-        senders[0].send(EventData(b"Receiving only a single event"), partition_key='0')
+        await senders[0].send(EventData(b"Receiving only a single event"), partition_key='0')
         await asyncio.sleep(10)
         assert on_event.called is True
 
     await task
+    for s in senders:
+        await s.close()
+    await producer_client.close()
 
 
 @pytest.mark.parametrize("position, inclusive, expected_result",
diff --git a/sdk/eventhub/azure-eventhub/tests/livetest/synctests/test_consumer_client.py b/sdk/eventhub/azure-eventhub/tests/livetest/synctests/test_consumer_client.py
index 5a2dca789f37..8da5ddeb6ead 100644
--- a/sdk/eventhub/azure-eventhub/tests/livetest/synctests/test_consumer_client.py
+++ b/sdk/eventhub/azure-eventhub/tests/livetest/synctests/test_consumer_client.py
@@ -17,7 +17,7 @@ def test_receive_no_partition(connstr_senders):
 
     def on_event(partition_context, event):
         on_event.received += 1
-        partition_context.update_checkpoint(event, fake_kwarg="arg")    # ignores fake_kwarg
+        partition_context.update_checkpoint(event)
         on_event.namespace = partition_context.fully_qualified_namespace
         on_event.eventhub_name = partition_context.eventhub_name
         on_event.consumer_group = partition_context.consumer_group