diff --git a/tests/fpgadataflow/test_fpgadataflow_convinputgenerator.py b/tests/fpgadataflow/test_fpgadataflow_convinputgenerator.py
index 0b7df33796d422aa260edb8a83c86cf907044b90..103d3fc5c78515378c9b9b4e2ec62ec54f514991 100644
--- a/tests/fpgadataflow/test_fpgadataflow_convinputgenerator.py
+++ b/tests/fpgadataflow/test_fpgadataflow_convinputgenerator.py
@@ -5,7 +5,7 @@ import finn.core.onnx_exec as oxe
 from finn.core.datatype import DataType
 from finn.core.modelwrapper import ModelWrapper
 from finn.core.utils import gen_finn_dt_tensor
-# from finn.transformation.fpgadataflow.cleanup import CleanUp
+from finn.transformation.fpgadataflow.cleanup import CleanUp
 from finn.transformation.fpgadataflow.codegen import CodeGen
 from finn.transformation.fpgadataflow.compile import Compile
 
@@ -89,3 +89,5 @@ def test_fpgadataflow_slidingwindow():
     y_produced = oxe.execute_onnx(model, input_dict)["outp"]
     print(x)
     print(y_produced)
+
+    model = model.transform(CleanUp())