diff --git a/vla_arena/vla_arena/bddl_files/distractor_dynamic_distractors/level_2/pick_up_the_carrot_and_put_it_on_the_bowl.bddl b/vla_arena/vla_arena/bddl_files/distractor_dynamic_distractors/level_2/pick_up_the_carrot_and_put_it_on_the_bowl.bddl
index 2f8cfb7b..b5e77da4 100644
--- a/vla_arena/vla_arena/bddl_files/distractor_dynamic_distractors/level_2/pick_up_the_carrot_and_put_it_on_the_bowl.bddl
+++ b/vla_arena/vla_arena/bddl_files/distractor_dynamic_distractors/level_2/pick_up_the_carrot_and_put_it_on_the_bowl.bddl
@@ -5,11 +5,11 @@
   (:regions
     (target_region
         (:target main_table)
-        (:ranges ((0.07 -0.25 0.08 -0.24)))
+        (:ranges ((0.07 -0.27 0.08 -0.26)))
     )
     (object_region
         (:target main_table)
-        (:ranges ((-0.29 0.22 -0.27 0.24)))
+        (:ranges ((-0.29 0.22 -0.28 0.24)))
     )
     (moving_region_1
         (:target main_table)
diff --git a/vla_arena/vla_arena/envs/__init__.py b/vla_arena/vla_arena/envs/__init__.py
index e6fe5e92..b013d463 100644
--- a/vla_arena/vla_arena/envs/__init__.py
+++ b/vla_arena/vla_arena/envs/__init__.py
@@ -18,5 +18,5 @@
 from .problems import *
 from .robots import *
 
-# from .env_wrapper import OffScreenRenderEnv, SegmentationRenderEnv
+from .env_wrapper import OffScreenRenderEnv, SegmentationRenderEnv
 from .venv import DummyVectorEnv, SubprocVectorEnv
diff --git a/vla_arena/vla_arena/init_files/distractor_dynamic_distractors/level_2/pick_up_the_carrot_and_put_it_on_the_bowl.pruned_init b/vla_arena/vla_arena/init_files/distractor_dynamic_distractors/level_2/pick_up_the_carrot_and_put_it_on_the_bowl.pruned_init
index 2c9820f1..81a5d8fa 100644
Binary files a/vla_arena/vla_arena/init_files/distractor_dynamic_distractors/level_2/pick_up_the_carrot_and_put_it_on_the_bowl.pruned_init and b/vla_arena/vla_arena/init_files/distractor_dynamic_distractors/level_2/pick_up_the_carrot_and_put_it_on_the_bowl.pruned_init differ
diff --git a/vla_arena/vla_arena/init_files/extrapolation_preposition_combinations/level_0/pick_the_tomato_next_to_the_cereal_and_place_it_on_the_porcelain_bowl_between_the_cabinet_and_the_cutting_board.pruned_init b/vla_arena/vla_arena/init_files/extrapolation_preposition_combinations/level_0/pick_the_tomato_next_to_the_cereal_and_place_it_on_the_porcelain_bowl_between_the_cabinet_and_the_cutting_board.pruned_init
index 6d650f55..c08b2340 100644
Binary files a/vla_arena/vla_arena/init_files/extrapolation_preposition_combinations/level_0/pick_the_tomato_next_to_the_cereal_and_place_it_on_the_porcelain_bowl_between_the_cabinet_and_the_cutting_board.pruned_init and b/vla_arena/vla_arena/init_files/extrapolation_preposition_combinations/level_0/pick_the_tomato_next_to_the_cereal_and_place_it_on_the_porcelain_bowl_between_the_cabinet_and_the_cutting_board.pruned_init differ