diff --git a/src/training/metrics/model_salience.py b/src/training/metrics/model_salience.py
index 2d6dfa3..64185e8 100644
--- a/src/training/metrics/model_salience.py
+++ b/src/training/metrics/model_salience.py
@@ -2,7 +2,7 @@
 import torch
 from transformers import AutoTokenizer, AutoModelForMaskedLM
 
-from utils_masking import NonStopMasker
+from .utils_masking import NonStopMasker
 
 
 def unfold(sent_toks, make_tensor=True):