diff --git a/py/image_mask_scale_as.py b/py/image_mask_scale_as.py
index 19948ee..0638266 100644
--- a/py/image_mask_scale_as.py
+++ b/py/image_mask_scale_as.py
@@ -1,6 +1,6 @@
 import torch
 from PIL import Image
-from .imagefunc import AnyType, log, tensor2pil, pil2tensor, fit_resize_image
+from .imagefunc import AnyType, log, tensor2pil, pil2tensor, image2mask, fit_resize_image