diff --git a/tests/test_brevitas_export.py b/tests/test_brevitas_export.py
index d0df91f74350bd11f9a1d2309e7aa6c6fe16b161..23d8af3d1a8e466e21be64a2d46b50241dcb8850 100644
--- a/tests/test_brevitas_export.py
+++ b/tests/test_brevitas_export.py
@@ -95,7 +95,7 @@ def test_brevitas_to_onnx_export():
         int_weights_onnx = nph.to_array(model.graph.node[13].attribute[0].t)
         assert (int_weights_onnx == int_weights_pytorch).all()
         assert model.graph.node[12].attribute[0].name == "activation_qnt"
-        assert model.graph.node[12].attribute[0].s.decode("utf-8") == "1"
+        assert model.graph.node[12].attribute[0].s.decode("utf-8") == "BIPOLAR"
         assert model.graph.node[14].attribute[1].name == "weight_qnt"
-        assert model.graph.node[14].attribute[1].s.decode("utf-8") == "1"
+        assert model.graph.node[14].attribute[1].s.decode("utf-8") == "BIPOLAR"
         os.remove(export_onnx_path)