diff --git a/ppdet/data/transform/operators.py b/ppdet/data/transform/operators.py
index 721f6d033b6a8f2539dea133158368165d7efee9..4ccd292ef1ef698d4deee0f40fca624caab74305 100644
--- a/ppdet/data/transform/operators.py
+++ b/ppdet/data/transform/operators.py
@@ -322,7 +322,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):