diff --git a/mace/python/tools/converter_tool/transformer.py b/mace/python/tools/converter_tool/transformer.py
index 657d344579c2f305a74d0120bba4e0159368aa4c..ca0d3e2a5109fe49c00faa5b92265ff378be318e 100644
--- a/mace/python/tools/converter_tool/transformer.py
+++ b/mace/python/tools/converter_tool/transformer.py
@@ -1786,7 +1786,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: