diff --git a/tests/test_streamline.py b/tests/test_streamline.py
index 75b0301072bd603e0c25357fa022b66d1df4e467..d0e83e8b5b75cf33b2b1cdedda46feee38ef339a 100644
--- a/tests/test_streamline.py
+++ b/tests/test_streamline.py
@@ -68,7 +68,7 @@ def test_streamline_lfc_w1a1():
         model = model.transform(InferDataTypes())
         produced_ctx = oxe.execute_onnx(model, input_dict, True)
         produced = produced_ctx[model.graph.output[0].name]
-        # model.save("%d-%s.onnx" % (trn_ind, trn.__name__))
+        # model.save("%d-%s.onnx" % (trn_ind, trn.__class__.__name__))
         assert np.isclose(expected, produced, atol=1e-3).all()
         trn_ind += 1
     os.remove(export_onnx_path)