diff --git a/tb/axis_replay_buffer.py b/tb/axis_replay_buffer.py
index d318f78..1911da9 100644
--- a/tb/axis_replay_buffer.py
+++ b/tb/axis_replay_buffer.py
@@ -42,11 +42,11 @@ async def recv_packet(signals, packet, last=None):
 
     for i, val in enumerate(packet):
         while not signals['valid'].value or not signals['ready'].value:
-            await RisingEdge(signals['clk'])
+            await FallingEdge(signals['clk'])
         assert signals['data'].value == val
         if 'last' in signals:
             assert signals['last'].value == (i == last - 1)
-        await RisingEdge(signals['clk'])
+        await FallingEdge(signals['clk'])
 
 @timeout(30, 'us')
 async def test_replay(buf, in_ratio, out_ratio):
diff --git a/tb/util.py b/tb/util.py
index 874f4ab..bcf9328 100644
--- a/tb/util.py
+++ b/tb/util.py
@@ -150,9 +150,9 @@ async def ClockEnable(clk, ce, ratio):
         return
 
     while True:
-        await ClockCycles(clk, 1, False)
+        await ClockCycles(clk, 1)
         ce.value = 0
-        await ClockCycles(clk, ratio - 1, False)
+        await ClockCycles(clk, ratio - 1)
         ce.value = 1
 
 # Adapted from https://stackoverflow.com/a/1630350/5086505