diff --git a/CAT/model/IRT.py b/CAT/model/IRT.py
index 5b828d0..1dfa576 100644
--- a/CAT/model/IRT.py
+++ b/CAT/model/IRT.py
@@ -13,7 +13,7 @@
 from CAT.dataset import AdapTestDataset, TrainDataset, Dataset
 from sklearn.metrics import accuracy_score
 from collections import namedtuple
-from .utils import StraightThrough
+from CAT.model.utils import StraightThrough
 SavedAction = namedtuple('SavedAction', ['log_prob', 'value'])
 
 class IRT(nn.Module):