diff --git a/tests/end2end/test_end2end_bnn_pynq.py b/tests/end2end/test_end2end_bnn_pynq.py
index 1fddc7c1c26a0ba04d5849809ccf59b0a926a509..bc14b9dee5de0b1c0a5d05292935ab91c1dd1553 100644
--- a/tests/end2end/test_end2end_bnn_pynq.py
+++ b/tests/end2end/test_end2end_bnn_pynq.py
@@ -206,8 +206,8 @@ def fold_cnv_small(model):
     fc_layers = model.get_nodes_by_op_type("StreamingFCLayer_Batch")
     # each tuple is (PE, SIMD) for a layer
     folding = [
-        (8, 3, "auto"),
-        (16, 16, "auto"),
+        (8, 3, "distributed"),
+        (16, 16, "distributed"),
         (8, 16, "auto"),
         (8, 16, "block"),
         (4, 8, "auto"),