diff --git a/moleculib/protein/dataset.py b/moleculib/protein/dataset.py
index 0fd07e3..1825f9c 100644
--- a/moleculib/protein/dataset.py
+++ b/moleculib/protein/dataset.py
@@ -490,7 +490,7 @@ def __init__(
         split: str = "train",
         tau: int = 1,
     ):
-        base = "/mas/projects/molecularmachines/db/timewarp/"
+        base = "/mas/projects/molecularmachines/db/timewarp2/"
         self.base_path = os.path.join(base, dataset, split)
         self.counter = 0
         self.tau = tau