diff --git a/src/finn/transformation/fpgadataflow/insert_tlastmarker.py b/src/finn/transformation/fpgadataflow/insert_tlastmarker.py
index d174f3acb7bee8cbe2c8ed76a1251ffc4cde0417..ed8b96852171d989817612ffddd2879c4e23d330 100644
--- a/src/finn/transformation/fpgadataflow/insert_tlastmarker.py
+++ b/src/finn/transformation/fpgadataflow/insert_tlastmarker.py
@@ -39,6 +39,8 @@ class InsertTLastMarker(Transformation):
                 [graph_out_name],
                 NumIters=num_iters,
                 StreamWidth=stream_width,
+                domain="finn",
+                backend="fpgadataflow",
             )
             model.graph.node.append(tlast_node)
             return (model, True)
diff --git a/tests/fpgadataflow/test_create_dataflow_partition.py b/tests/fpgadataflow/test_create_dataflow_partition.py
index ed5b95654c6501b177621c3f4fdad12be109e776..af3dc5816ab354dc71320507a8589d3c7ca1fb4a 100644
--- a/tests/fpgadataflow/test_create_dataflow_partition.py
+++ b/tests/fpgadataflow/test_create_dataflow_partition.py
@@ -36,6 +36,7 @@ def test_dataflow_partition_tlastmarker():
     model = ModelWrapper(model_path)
     model = model.transform(InsertTLastMarker())
     assert model.graph.node[-1].op_type == "TLastMarker"
+    assert model.graph.node[-1].domain == "finn"
     tl_node = getCustomOp(model.graph.node[-1])
     assert tl_node.get_nodeattr("NumIters") == 1
     assert tl_node.get_nodeattr("StreamWidth") == 320