diff --git a/tests/test_wave_sim.py b/tests/test_wave_sim.py
index 63bb226..b7b6676 100644
--- a/tests/test_wave_sim.py
+++ b/tests/test_wave_sim.py
@@ -41,11 +41,10 @@ def test_xnor2_delays():
 
 def test_nand_delays():
     op = (sim.NAND4, 4, 0, 1, 2, 3, -1, 0, 0)
-    #op = (0b0111, 4, 0, 1)
     c = np.full((5*16, 1), TMAX, dtype=np.float32)  # 5 waveforms of capacity 16
     c_locs = np.zeros((5,), dtype='int')
     c_caps = np.zeros((5,), dtype='int')
-    ebuf = np.zeros((4, 1, 2), dtype=np.int32)
+    ebuf = np.zeros((5, 1, 2), dtype=np.int32)
 
     for i in range(5): c_locs[i], c_caps[i] = i*16, 16  # 1:1 mapping