diff --git a/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py b/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py
index 3809e327943832571a1bde6a53a0a6e7fbd13bdd..44767220d06a976ef86186b69765c2b3e44b3ac2 100644
--- a/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py
+++ b/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py
@@ -383,7 +383,7 @@ class QuantizationTransformPass(object):
             data_type = 'float64' if var_node.dtype(
             ) == core.VarDesc.VarType.FP64 else 'float32'
             _init_var_node(
-                scale_in_node,
+                state_in_node,
                 np.ones(
                     [1], dtype=data_type),
                 self._scope,