diff --git a/onnx/test/shape_inference_test.py b/onnx/test/shape_inference_test.py
index a74e8485dd4..da6e9d374c1 100644
--- a/onnx/test/shape_inference_test.py
+++ b/onnx/test/shape_inference_test.py
@@ -978,7 +978,7 @@ def test_resize_size(self, _, version) -> None:
             )
 
     @parameterized.expand(all_versions_for("RMSNormalization"))
-    def test_rms_normalization(self) -> None:
+    def test_rms_normalization(self _, version) -> None:
         graph = self._make_graph(
             [
                 ("X", TensorProto.FLOAT, ("N", "C", "H", "W")),