diff --git a/tests/fpgadataflow/test_create_dataflow_partition.py b/tests/fpgadataflow/test_create_dataflow_partition.py
index af3dc5816ab354dc71320507a8589d3c7ca1fb4a..5b78a2000e4327440cadfa81392e66742e07634f 100644
--- a/tests/fpgadataflow/test_create_dataflow_partition.py
+++ b/tests/fpgadataflow/test_create_dataflow_partition.py
@@ -40,6 +40,7 @@ def test_dataflow_partition_tlastmarker():
     tl_node = getCustomOp(model.graph.node[-1])
     assert tl_node.get_nodeattr("NumIters") == 1
     assert tl_node.get_nodeattr("StreamWidth") == 320
+    assert tl_node.get_nodeattr("ElemWidth") == 32
     model.save(build_dir + "/test_dataflow_partition_tlastmarker.onnx")
     model = model.transform(InsertTLastMarker())
     model.save(build_dir + "/test_dataflow_partition_tlastmarker2.onnx")
diff --git a/tests/fpgadataflow/test_fpgadataflow_ip_stitch.py b/tests/fpgadataflow/test_fpgadataflow_ip_stitch.py
index 32de9b382c0c95d7411a5e17ec78e6c4ea876673..3f23eb095109f1d0b087a20a49690ce8a09882df 100644
--- a/tests/fpgadataflow/test_fpgadataflow_ip_stitch.py
+++ b/tests/fpgadataflow/test_fpgadataflow_ip_stitch.py
@@ -75,6 +75,7 @@ def create_one_fc_model():
         domain="finn",
         backend="fpgadataflow",
         NumIters=1,
+        ElemWidth=odt.bitwidth(),
         StreamWidth=odt.bitwidth() * m,
     )
 
@@ -164,6 +165,7 @@ def create_two_fc_model():
         backend="fpgadataflow",
         NumIters=m,
         StreamWidth=2,
+        ElemWidth=odt.bitwidth(),
     )
 
     graph = helper.make_graph(