diff --git a/tests/test_custom_onnx_exec.py b/tests/test_custom_onnx_exec.py
index f355653311421f41ca5d1c7687d3571b14e6264c..a8b0560b24473212cf072d6534dc6296a3016d65 100644
--- a/tests/test_custom_onnx_exec.py
+++ b/tests/test_custom_onnx_exec.py
@@ -1,6 +1,4 @@
 import numpy as np
-# import onnx
-# from onnx import AttributeProto, GraphProto, TensorProto, helper
 from onnx import TensorProto, helper
 
 import finn.core.execute_custom_node as ex_cu_node