diff --git a/tests/data/action.memmap b/tests/data/pusht/action.memmap
similarity index 100%
rename from tests/data/action.memmap
rename to tests/data/pusht/action.memmap
diff --git a/tests/data/episode.memmap b/tests/data/pusht/episode.memmap
similarity index 100%
rename from tests/data/episode.memmap
rename to tests/data/pusht/episode.memmap
diff --git a/tests/data/frame_id.memmap b/tests/data/pusht/frame_id.memmap
similarity index 100%
rename from tests/data/frame_id.memmap
rename to tests/data/pusht/frame_id.memmap
diff --git a/tests/data/meta.json b/tests/data/pusht/meta.json
similarity index 100%
rename from tests/data/meta.json
rename to tests/data/pusht/meta.json
diff --git a/tests/data/next/done.memmap b/tests/data/pusht/next/done.memmap
similarity index 100%
rename from tests/data/next/done.memmap
rename to tests/data/pusht/next/done.memmap
diff --git a/tests/data/next/meta.json b/tests/data/pusht/next/meta.json
similarity index 100%
rename from tests/data/next/meta.json
rename to tests/data/pusht/next/meta.json
diff --git a/tests/data/next/observation/image.memmap b/tests/data/pusht/next/observation/image.memmap
similarity index 100%
rename from tests/data/next/observation/image.memmap
rename to tests/data/pusht/next/observation/image.memmap
diff --git a/tests/data/next/observation/meta.json b/tests/data/pusht/next/observation/meta.json
similarity index 100%
rename from tests/data/next/observation/meta.json
rename to tests/data/pusht/next/observation/meta.json
diff --git a/tests/data/next/observation/state.memmap b/tests/data/pusht/next/observation/state.memmap
similarity index 100%
rename from tests/data/next/observation/state.memmap
rename to tests/data/pusht/next/observation/state.memmap
diff --git a/tests/data/next/reward.memmap b/tests/data/pusht/next/reward.memmap
similarity index 100%
rename from tests/data/next/reward.memmap
rename to tests/data/pusht/next/reward.memmap
diff --git a/tests/data/next/success.memmap b/tests/data/pusht/next/success.memmap
similarity index 100%
rename from tests/data/next/success.memmap
rename to tests/data/pusht/next/success.memmap
diff --git a/tests/data/observation/image.memmap b/tests/data/pusht/observation/image.memmap
similarity index 100%
rename from tests/data/observation/image.memmap
rename to tests/data/pusht/observation/image.memmap
diff --git a/tests/data/observation/meta.json b/tests/data/pusht/observation/meta.json
similarity index 100%
rename from tests/data/observation/meta.json
rename to tests/data/pusht/observation/meta.json
diff --git a/tests/data/observation/state.memmap b/tests/data/pusht/observation/state.memmap
similarity index 100%
rename from tests/data/observation/state.memmap
rename to tests/data/pusht/observation/state.memmap
diff --git a/tests/data/stats.pth b/tests/data/pusht/stats.pth
similarity index 100%
rename from tests/data/stats.pth
rename to tests/data/pusht/stats.pth