diff --git a/src/timesfm.py b/src/timesfm.py
index 7dc9d0a..07a865c 100644
--- a/src/timesfm.py
+++ b/src/timesfm.py
@@ -35,7 +35,7 @@
 from praxis import pytypes
 from praxis.layers import normalizations
 from praxis.layers import transformers
-import patched_decoder
+from . import patched_decoder
 from utilsforecast.processing import make_future_dataframe
 
 instantiate = base_hyperparams.instantiate