diff --git a/common/research/dataset/twitch/robots_views_extractor.py b/common/research/dataset/twitch/robots_views_extractor.py
index 4b0baa61ec661a1459ad8582af7256f923e11e34..c6f7ed4295d7def38d9f80952200c8673476f6c4 100644
--- a/common/research/dataset/twitch/robots_views_extractor.py
+++ b/common/research/dataset/twitch/robots_views_extractor.py
@@ -6,7 +6,7 @@ import numpy as np
 from tqdm import tqdm
 
 from research.constants import TWITCH_DSET
-from research.dataset.twitch.new_vision import is_image_from_robot_view
+from research.dataset.twitch.mask_detector import is_image_from_robot_view
 from research.dataset.twitch.video_frame_generator import VideoFrameGenerator
 
 RES_DIR: Path = TWITCH_DSET / 'robots-views'