diff --git a/test/scheduler/test_scheduler.py b/test/scheduler/test_scheduler.py
index ea09270dc..fbfb7eb7a 100644
--- a/test/scheduler/test_scheduler.py
+++ b/test/scheduler/test_scheduler.py
@@ -207,8 +207,8 @@ def make_queue_process(
 
         Parameters
         ----------
-        io : AsyncTestbenchIO
-            AsyncTestbenchIO to call with items gathered from `input_queues`.
+        io : TestbenchIO
+            TestbenchIO to call with items gathered from `input_queues`.
         input_queues : deque[dict], optional
             Queue of dictionaries containing fields and values of a record to call
             `io` with. Different fields may be split across multiple queues.
diff --git a/transactron/testing/infrastructure.py b/transactron/testing/infrastructure.py
index 83530033d..229326816 100644
--- a/transactron/testing/infrastructure.py
+++ b/transactron/testing/infrastructure.py
@@ -314,7 +314,7 @@ def run_simulation(self, module: HasElaborate, max_cycles: float = 10e4, add_tra
 
     async def tick(self, sim: AnySimulatorContext, cycle_cnt: int = 1):
         """
-        Yields for the given number of cycles.
+        Waits for the given number of cycles.
         """
         for _ in range(cycle_cnt):
             await sim.tick()