diff --git a/fog_x/dataset.py b/fog_x/dataset.py
index cd59771..f20d343 100644
--- a/fog_x/dataset.py
+++ b/fog_x/dataset.py
@@ -412,7 +412,7 @@ def load_rtx_episodes_local(
         # this is only required if rtx format is used
         import tensorflow_datasets as tfds
 
-        b = tfds.builder(path)
+        b = tfds.builder_from_directory(path)
         self._build_rtx_episodes_from_tfds_builder(
             b,
             split=split,