diff --git a/mace/python/tools/converter_tool/onnx_converter.py b/mace/python/tools/converter_tool/onnx_converter.py
index 70494ae7e7aa9adbcf67656648512c914f1311b7..ee7a523cc78a4f2fee7f945742040795ca14e01d 100644
--- a/mace/python/tools/converter_tool/onnx_converter.py
+++ b/mace/python/tools/converter_tool/onnx_converter.py
@@ -1395,8 +1395,8 @@ class OnnxConverter(base_converter.ConverterInterface):
             op.type = MaceOp.Squeeze.name
             axis_arg = op.arg.add()
             axis_arg.name = MaceKeyword.mace_axis_str
-            if 'axis' in node.attrs:
-                axis_value = node.attrs['axis']
+            if 'axes' in node.attrs:
+                axis_value = node.attrs['axes']
             else:
                 axis_value = []
             axis_arg.ints.extend(axis_value)