diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py
index 0c918cf677b80b2deb64106159e11ac97cd2fd02..9d1d5fe0932ea8a53e28bc18a776a430a53e9ef4 100644
--- a/python/paddle/fluid/layers/nn.py
+++ b/python/paddle/fluid/layers/nn.py
@@ -10654,8 +10654,8 @@ def tree_conv(nodes_vector,
     return helper.append_activation(pre_activation)
 
 
-from .control_flow import equal
 from .ops import square
+from .control_flow import equal
 
 
 def npair_loss(anchor, positive, labels, l2_reg=0.002):