diff --git a/ppdet/modeling/layers.py b/ppdet/modeling/layers.py
index 671907898050904f5a09a52e11bec7e51d6851de..424e9017dccef7317c94fa3584cad841237e7109 100644
--- a/ppdet/modeling/layers.py
+++ b/ppdet/modeling/layers.py
@@ -401,9 +401,8 @@ class YOLOBox(object):
         self.clip_bbox = clip_bbox
 
     def __call__(self, x, img_size, anchors, num_classes, stage=0):
-        outs = fluid.layers.yolo_box(x, img_size, anchors, num_classes,
-                                     self.conf_thresh, self.downsample_ratio //
-                                     2**stage, self.clip_bbox)
+        outs = ops.yolo_box(x, img_size, anchors, num_classes, self.conf_thresh,
+                            self.downsample_ratio // 2**stage, self.clip_bbox)
         return outs