diff --git a/.github/unittest/linux_libs/scripts_habitat/run_test.sh b/.github/unittest/linux_libs/scripts_habitat/run_test.sh
index 0cba133ae8e..f790e757d3d 100755
--- a/.github/unittest/linux_libs/scripts_habitat/run_test.sh
+++ b/.github/unittest/linux_libs/scripts_habitat/run_test.sh
@@ -45,8 +45,8 @@ conda deactivate && conda activate ./env
 mkdir data
 git lfs update
 
-#python -m habitat_sim.utils.datasets_download --uids rearrange_dataset_v1 rearrange_pick_dataset_v0 rearrange_task_assets --data-path ./data
-python -m habitat_sim.utils.datasets_download --uids rearrange_task_assets --data-path ./data --no-prune
+python -m habitat_sim.utils.datasets_download --uids rearrange_pick_dataset_v0 rearrange_task_assets --data-path ./data --no-prune
+#python -m habitat_sim.utils.datasets_download --uids rearrange_task_assets --data-path ./data --no-prune
 
 python -c "import habitat;import habitat.gym"
 python -c """from torchrl.envs.libs.habitat import HabitatEnv