diff --git a/ppdet/data/transform/operators.py b/ppdet/data/transform/operators.py
index b706a43ba716375c86cd5c8c05b5c27301c88f80..8c6a65e7bd02dd169e392447529ee06c488852e4 100644
--- a/ppdet/data/transform/operators.py
+++ b/ppdet/data/transform/operators.py
@@ -325,7 +325,7 @@ class RandomErasingImage(BaseOperator):
         self.higher = higher
         self.aspect_ratio = aspect_ratio
 
-    def apply(self, sample):
+    def apply(self, sample, context=None):
         gt_bbox = sample['gt_bbox']
         im = sample['image']
         if not isinstance(im, np.ndarray):