diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py
index f6092750e7aa803ec39ba75e06604b93c4bd2a10..a546864f5c3f95abdb35b522fa08387bc7eca288 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
+from .ops import square
 
 
 def npair_loss(anchor, positive, labels, l2_reg=0.002):