diff --git a/utils/timefeatures.py b/utils/timefeatures.py
index 2708026..c6ba0b6 100644
--- a/utils/timefeatures.py
+++ b/utils/timefeatures.py
@@ -35,7 +35,7 @@ class HourOfDay(TimeFeature):
 
 class DayOfWeek(TimeFeature):
     """Hour of day encoded as value between [-0.5, 0.5]"""
-    def __call__(self, index: pd.DatetimeIndex) -> np.ndarray:
+    def __call__(self, index: pd.DatetimeIndex) -> mnp.ndarray:
         return index.dayofweek / 6.0 - 0.5
 
 class DayOfMonth(TimeFeature):