diff --git a/data/envs/metaworld/generate_dataset_all.sh b/data/envs/metaworld/generate_dataset_all.sh
index acfe879..0185b2b 100755
--- a/data/envs/metaworld/generate_dataset_all.sh
+++ b/data/envs/metaworld/generate_dataset_all.sh
@@ -2,8 +2,6 @@
 
 ENVS=(
     assembly
-    basketball
-    bin-picking
     box-close
     button-press-topdown
     button-press-topdown-wall
@@ -11,9 +9,7 @@ ENVS=(
     button-press-wall
     coffee-button
     coffee-pull
-    coffee-push
     dial-turn
-    disassemble
     door-close
     door-lock
     door-open
@@ -22,29 +18,15 @@ ENVS=(
     drawer-open
     faucet-close
     faucet-open
-    hammer
     hand-insert
     handle-press-side
     handle-press
     handle-pull-side
     handle-pull
     lever-pull
-    peg-insert-side
-    peg-unplug-side
-    pick-out-of-hole
-    pick-place
-    pick-place-wall
-    plate-slide-back-side
-    plate-slide-back
-    plate-slide-side
-    plate-slide
     push-back
     push
     push-wall
-    reach
-    reach-wall
-    shelf-place
-    soccer
     stick-pull
     stick-push
     sweep-into
@@ -54,6 +36,6 @@ ENVS=(
 )
 
 for ENV in "${ENVS[@]}"; do
-    python -m sample_factory.huggingface.load_from_hub -r qgallouedec/$ENV-v2
+    # python -m sample_factory.huggingface.load_from_hub -r qgallouedec/$ENV-v2
     python generate_dataset.py --env $ENV-v2 --experiment $ENV-v2 --train_dir=./train_dir
 done
diff --git a/data/envs/metaworld/train.py b/data/envs/metaworld/train.py
index 095414e..0ea5bde 100644
--- a/data/envs/metaworld/train.py
+++ b/data/envs/metaworld/train.py
@@ -25,7 +25,7 @@ def override_defaults(parser: argparse.ArgumentParser) -> argparse.ArgumentParse
         num_workers=8,
         num_envs_per_worker=8,
         worker_num_splits=2,
-        train_for_env_steps=10_000_000,
+        train_for_env_steps=30_000_000,
         encoder_mlp_layers=[64, 64],
         env_frameskip=1,
         nonlinearity="tanh",
diff --git a/data/envs/metaworld/train_all.sh b/data/envs/metaworld/train_all.sh
index dbf328a..67ab9a0 100755
--- a/data/envs/metaworld/train_all.sh
+++ b/data/envs/metaworld/train_all.sh
@@ -1,56 +1,8 @@
 #!/bin/bash
 
 ENVS=(
-    assembly
-    basketball
-    bin-picking
-    box-close
-    button-press-topdown
-    button-press-topdown-wall
-    button-press
-    button-press-wall
-    coffee-button
-    coffee-pull
-    coffee-push
-    dial-turn
     disassemble
-    door-close
-    door-lock
-    door-open
-    door-unlock
-    drawer-close
-    drawer-open
-    faucet-close
-    faucet-open
-    hammer
-    hand-insert
-    handle-press-side
-    handle-press
-    handle-pull-side
-    handle-pull
-    lever-pull
     peg-insert-side
-    peg-unplug-side
-    pick-out-of-hole
-    pick-place
-    pick-place-wall
-    plate-slide-back-side
-    plate-slide-back
-    plate-slide-side
-    plate-slide
-    push-back
-    push
-    push-wall
-    reach
-    reach-wall
-    shelf-place
-    soccer
-    stick-pull
-    stick-push
-    sweep-into
-    sweep
-    window-close
-    window-open
 )
 
 for ENV in "${ENVS[@]}"; do
diff --git a/gia/eval/rl/gym_evaluator.py b/gia/eval/rl/gym_evaluator.py
index f8531ee..44f5f91 100644
--- a/gia/eval/rl/gym_evaluator.py
+++ b/gia/eval/rl/gym_evaluator.py
@@ -1,7 +1,7 @@
 import gym
 from gym.vector.vector_env import VectorEnv
 
-from gia.eval.mappings import TASK_TO_ENV_MAPPING
+# from gia.eval.mappings import TASK_TO_ENV_MAPPING
 from gia.eval.rl.rl_evaluator import RLEvaluator