diff --git a/node_wrappers/mediapipe_vision/image_segmentation/image_segmentation_masking.py b/node_wrappers/mediapipe_vision/image_segmentation/image_segmentation_masking.py
index e643a66..e4ee1e2 100644
--- a/node_wrappers/mediapipe_vision/image_segmentation/image_segmentation_masking.py
+++ b/node_wrappers/mediapipe_vision/image_segmentation/image_segmentation_masking.py
@@ -1,5 +1,4 @@
 import torch
-import numpy as np
 import logging