diff --git a/calyx-py/test/correctness/pifo.py b/calyx-py/test/correctness/pifo.py
index c9d36626e6..a7a670d00c 100644
--- a/calyx-py/test/correctness/pifo.py
+++ b/calyx-py/test/correctness/pifo.py
@@ -104,16 +104,16 @@ def insert_pifo(prog, name, queue_l, queue_r, boundary):
     hot = pifo.reg("hot", 1)
 
     # Some equality checks.
-    hot_eq_0 = util.insert_eq(pifo, hot.out, 0, "hot_eq_0", 1)
-    hot_eq_1 = util.insert_eq(pifo, hot.out, 1, "hot_eq_1", 1)
-    flow_eq_0 = util.insert_eq(pifo, flow.out, 0, "flow_eq_0", 1)
-    flow_eq_1 = util.insert_eq(pifo, flow.out, 1, "flow_eq_1", 1)
-    len_eq_0 = util.insert_eq(pifo, len.out, 0, "len_eq_0", 32)
-    len_eq_10 = util.insert_eq(pifo, len.out, 10, "len_eq_10", 32)
-    cmd_eq_0 = util.insert_eq(pifo, cmd, 0, "cmd_eq_0", 32)
-    cmd_eq_1 = util.insert_eq(pifo, cmd, 1, "cmd_eq_1", 32)
+    hot_eq_0 = util.insert_eq(pifo, hot.out, 0, 1, "hot_eq_0")
+    hot_eq_1 = util.insert_eq(pifo, hot.out, 1, 1, "hot_eq_1")
+    flow_eq_0 = util.insert_eq(pifo, flow.out, 0, 1, "flow_eq_0")
+    flow_eq_1 = util.insert_eq(pifo, flow.out, 1, 1, "flow_eq_1")
+    len_eq_0 = util.insert_eq(pifo, len.out, 0, 32, "len_eq_0")
+    len_eq_10 = util.insert_eq(pifo, len.out, 10, 32, "len_eq_10")
+    cmd_eq_0 = util.insert_eq(pifo, cmd, 0, 32, "cmd_eq_0")
+    cmd_eq_1 = util.insert_eq(pifo, cmd, 1, 32, "cmd_eq_1")
     cmd_gt_1 = util.insert_gt(pifo, cmd, 1, "cmd_gt_1", 32)
-    err_eq_0 = util.insert_eq(pifo, err.out, 0, "err_eq_0", 1)
+    err_eq_0 = util.insert_eq(pifo, err.out, 0, 1, "err_eq_0")
     err_neq_0 = util.insert_neq(pifo, err.out, cb.const(1, 0), "err_neq_0", 1)
 
     flip_hot = util.insert_bitwise_flip_reg(pifo, hot, "flip_hot", 1)