diff --git a/caffe2fluid/kaffe/paddle/network.py b/caffe2fluid/kaffe/paddle/network.py
index 86c6b0e96be086ece88709c454b6871a983696c1..87d1c27a17bc774234b13476dab72a69eb99f820 100644
--- a/caffe2fluid/kaffe/paddle/network.py
+++ b/caffe2fluid/kaffe/paddle/network.py
@@ -528,14 +528,14 @@ class Network(object):
         if isinstance(input, list) and len(input) == 2:
             # for two tensor, here resets axis to 1. Maybe there is a bug for unkown case.
             axis = 1
-            output_shape = input[0].shape[axis:axis + num_axes]
+            bias_shape = input[0].shape[axis:axis + num_axes]
             scale_param = input[1]
             input = input[0]
         else:
-            output_shape = input.shape[axis:axis + num_axes]
+            bias_shape = input.shape[axis:axis + num_axes]
             param_attr = fluid.ParamAttr(name=prefix + 'scale')
             scale_param = fluid.layers.create_parameter(
-                shape=output_shape,
+                shape=bias_shape,
                 dtype=input.dtype,
                 name=name,
                 attr=param_attr,
@@ -548,7 +548,7 @@ class Network(object):
             axis=axis,
             name=self.get_unique_output_name(name, 'scale_mul'))
 
-        scale_shape = output_shape
+        scale_shape = bias_shape
         offset_attr = fluid.ParamAttr(name=prefix + 'offset')
         offset_param = fluid.layers.create_parameter(
             shape=scale_shape,