diff --git a/mace/python/tools/converter_tool/transformer.py b/mace/python/tools/converter_tool/transformer.py
index 6ce3beba976adffbfa1ed8895a050dba99fd5e97..1083e23545767725e2f4e0d9c394d790fd5d0dd3 100644
--- a/mace/python/tools/converter_tool/transformer.py
+++ b/mace/python/tools/converter_tool/transformer.py
@@ -198,7 +198,7 @@ class Transformer(base_converter.ConverterInterface):
                     op.output.extend([input_node.name])
                     output_shape = op.output_shape.add()
                     output_shape.dims.extend(input_node.shape)
-                    if input_node in self._consumers:
+                    if input_node.name in self._consumers:
                         if ConverterUtil.data_format(
                                 self._consumers[input_node.name][0]) \
                                 == DataFormat.NCHW: