diff --git a/tools/caffe2onnx/src/OPs/Log.py b/tools/caffe2onnx/src/OPs/Log.py
index 327d42879..f7212c975 100644
--- a/tools/caffe2onnx/src/OPs/Log.py
+++ b/tools/caffe2onnx/src/OPs/Log.py
@@ -31,7 +31,7 @@ def get_log_output_shape(input_shape):
 
 
 def create_log_node(layer, node_name, input_name, output_name, input_shape):
-    output_shape = get_log_output_shape(layer)
+    output_shape = get_log_output_shape(input_shape)
 
     node = Node.c2oNode(layer, node_name, 'Log', input_name, output_name, input_shape, output_shape)