diff --git a/mace/python/tools/converter_tool/transformer.py b/mace/python/tools/converter_tool/transformer.py
index 49cba5b82a9fc125dd25046e11d3059bafe34f75..86982a2ce97b367e72e0985e7c73671d31f897a0 100644
--- a/mace/python/tools/converter_tool/transformer.py
+++ b/mace/python/tools/converter_tool/transformer.py
@@ -1777,7 +1777,8 @@ class Transformer(base_converter.ConverterInterface):
         net = self._model
         for op in net.op:
             # whether to reshape fc output(default 4D)
-            if op.type == MaceOp.FullyConnected.name:
+            if op.type == MaceOp.FullyConnected.name and\
+                    op.output[0] in self._consumers:
                 consumers = self._consumers[op.output[0]]
                 op_output_shape = op.output_shape[0].dims[:]
                 for consumer in consumers: