diff --git a/icevision/models/inference_sahi.py b/icevision/models/inference_sahi.py
index 6a4a7b5a4..a350b88c3 100644
--- a/icevision/models/inference_sahi.py
+++ b/icevision/models/inference_sahi.py
@@ -11,7 +11,7 @@
 from icevision.tfms.albumentations import albumentations_adapter
 
 
-from sahi.model import DetectionModel
+from sahi.models.base import DetectionModel
 from sahi.prediction import ObjectPrediction
 from sahi.predict import get_sliced_prediction as sahi_get_sliced_prediction