diff --git a/augraphy/utilities/interop.py b/augraphy/utilities/interop.py
index cee458c..0c929e6 100644
--- a/augraphy/utilities/interop.py
+++ b/augraphy/utilities/interop.py
@@ -22,7 +22,7 @@ def __init__(self, augmentation, p=1):
         self.augmentation = augmentation
         super().__init__(p=p)
 
-    def __call__(self, image, layer=None, force=False):
+    def __call__(self, image, layer=None, mask=None, keypoints=None, bounding_boxes=None, force=False):
 
         image = image.copy()
         result = self.augmentation(image=image)