diff --git a/src/laion_clap/hook.py b/src/laion_clap/hook.py
index a7f23ea..0437001 100644
--- a/src/laion_clap/hook.py
+++ b/src/laion_clap/hook.py
@@ -9,8 +9,8 @@
 import torch
 import librosa
 from clap_module import create_model
-from training.data import get_audio_features
-from training.data import int16_to_float32, float32_to_int16
+from .training.data import get_audio_features
+from .training.data import int16_to_float32, float32_to_int16
 
 from transformers import RobertaTokenizer
 import wget