diff --git a/x2paddle/optimizer/tensorflow/prelu.py b/x2paddle/optimizer/tensorflow/prelu.py
index 6bb8322b654997299f39b8686734e65ee59e5767..30b298d0b8ae70f92ec63f1f08ace63d41e9f9b3 100644
--- a/x2paddle/optimizer/tensorflow/prelu.py
+++ b/x2paddle/optimizer/tensorflow/prelu.py
@@ -118,6 +118,6 @@ class PReLUOpt:
                 graph.layers[transpose0.id] = transpose0
                 graph.layers[prelu.id] = prelu
                 graph.layers[transpose1.id] = transpose1
-                axis1 = graph.parameters[alpha.outputs[0]].shape[0]
-                graph.parameters[alpha.outputs[0]] = np.reshape(graph.parameters[alpha.outputs[0]], (1, axis1, 1, 1))
+                first_axis = graph.parameters[alpha.outputs[0]].shape[0]
+                graph.parameters[alpha.outputs[0]] = np.reshape(graph.parameters[alpha.outputs[0]], (1, first_axis, 1, 1))
         graph.build()
\ No newline at end of file