diff --git a/__init__.py b/__init__.py
index 2718f01..8275b16 100644
--- a/__init__.py
+++ b/__init__.py
@@ -16,6 +16,6 @@
 
 from __future__ import absolute_import
 
-from src.patched_decoder import PatchedTimeSeriesDecoder
-from src.timesfm import TimesFm
-from src.timesfm import freq_map
+from .src.patched_decoder import PatchedTimeSeriesDecoder
+from .src.timesfm import TimesFm
+from .src.timesfm import freq_map
diff --git a/src/timesfm.py b/src/timesfm.py
index 8850239..df21223 100644
--- a/src/timesfm.py
+++ b/src/timesfm.py
@@ -35,7 +35,7 @@ from praxis import py_utils
 from praxis import pytypes
 from praxis.layers import normalizations
 from praxis.layers import transformers
-from src import patched_decoder
+from . import patched_decoder
 from utilsforecast.processing import make_future_dataframe
 
 instantiate = base_hyperparams.instantiate