diff --git a/src/stimulus/learner/raytune_learner.py b/src/stimulus/learner/raytune_learner.py
index f9b594b..be2c2de 100644
--- a/src/stimulus/learner/raytune_learner.py
+++ b/src/stimulus/learner/raytune_learner.py
@@ -15,7 +15,7 @@
 from torch import nn, optim
 from torch.utils.data import DataLoader, Dataset
 
-from stimulus.data.experiments import EncoderLoader
+from stimulus.data.loaders import EncoderLoader
 from stimulus.data.handlertorch import TorchDataset
 from stimulus.learner.predict import PredictWrapper
 from stimulus.utils.generic_utils import set_general_seeds