diff --git a/tests/transformation/test_absorb_transp_into_flatten.py b/tests/transformation/test_absorb_transp_into_flatten.py
index fbfa15277717c554da01e38608601997407803b2..cbbb33b4606acf55ace662da0986105f8c456b39 100644
--- a/tests/transformation/test_absorb_transp_into_flatten.py
+++ b/tests/transformation/test_absorb_transp_into_flatten.py
@@ -57,9 +57,9 @@ def test_absorb_transp_into_flatten(perm, shape, ishape, data_layout):
     model = model.transform(InferDataLayouts())
     model = model.transform(GiveUniqueNodeNames())
     model = model.transform(GiveReadableTensorNames())
-    model.save("test.onnx")
+    # model.save("test.onnx")
     model_transformed = model.transform(AbsorbTransposeIntoFlatten())
-    model_transformed.save("test2.onnx")
+    # model_transformed.save("test2.onnx")
 
     # verify transformation
     inp_values = np.random.uniform(low=-1, high=1, size=tuple(ishape)).astype(
diff --git a/tests/transformation/test_topk_insert.py b/tests/transformation/test_topk_insert.py
index b85ed4aa6999faf751e535c1cc687d639c4eb74f..a18e63384150f140cb63ec7b438283eb4797266c 100644
--- a/tests/transformation/test_topk_insert.py
+++ b/tests/transformation/test_topk_insert.py
@@ -1,4 +1,4 @@
-# import os
+import os
 import onnx
 from finn.util.test import get_test_model_trained
 import brevitas.onnx as bo
@@ -57,4 +57,4 @@ def test_topk_insert(k):
     output_pysim_topk = output_pysim_topk.astype(np.int).flatten()
 
     assert np.array_equal(output_golden_topk, output_pysim_topk)
-    # os.remove(export_onnx_path)
+    os.remove(export_onnx_path)
diff --git a/tests/util/test_create.py b/tests/util/test_create.py
index 7173add35abf04a35c33b0ef10b42ffdb296a653..4e236978592b02e1c18b03aba56ff8b2369311a6 100644
--- a/tests/util/test_create.py
+++ b/tests/util/test_create.py
@@ -61,4 +61,4 @@ def test_hls_random_mlp_maker(bitwidth):
 
     ret = create.hls_random_mlp_maker(layer_spec)
     assert len(ret.graph.node) == 5
-    ret.save("mlp-%s.onnx" % str(bitwidth))
+    # ret.save("mlp-%s.onnx" % str(bitwidth))