diff --git a/lerobot/common/envs/utils.py b/lerobot/common/envs/utils.py
index 9bb9d81f..b34b77d8 100644
--- a/lerobot/common/envs/utils.py
+++ b/lerobot/common/envs/utils.py
@@ -115,7 +115,8 @@ def check_env_attributes_and_types(env: gym.vector.VectorEnv) -> None:
             )
 
 
-def infer_envs_task(env: gym.vector.VectorEnv, observation: dict[str, Any]) -> dict[str, Any]:
+def add_envs_task(env: gym.vector.VectorEnv, observation: dict[str, Any]) -> dict[str, Any]:
+    """ Adds task feature to the observation dict with respect to the first environment attribute. """
     if hasattr(env.envs[0], "task_description"):
         observation["task"] = env.call("task_description")
     elif hasattr(env.envs[0], "task"):