diff --git a/mace/python/tools/converter_tool/transformer.py b/mace/python/tools/converter_tool/transformer.py
index dfa92bf3eeca5d63127a0456dfa69b0aa9dd2905..60cfe88b2f33ec8d880f6c2cf17ddd948bf1f2ee 100644
--- a/mace/python/tools/converter_tool/transformer.py
+++ b/mace/python/tools/converter_tool/transformer.py
@@ -1174,7 +1174,7 @@ class Transformer(base_converter.ConverterInterface):
                 self.buffer_to_image(op, 0, OpenCLBufferType.WINOGRAD_FILTER)
             elif op.type == MaceOp.WinogradInverseTransform.name \
                     and len(op.input) >= 3:
-                self.buffer_to_image(op, 1, OpenCLBufferType.ARGUMENT)
+                self.buffer_to_image(op, 2, OpenCLBufferType.ARGUMENT)
             elif op.type == MaceOp.FullyConnected.name:
                 self.buffer_to_image(op, 1, OpenCLBufferType.WEIGHT_WIDTH)
                 if len(op.input) >= 3: