diff --git a/desktop_env/envs/desktop_env.py b/desktop_env/envs/desktop_env.py
index fee3f37..7dd70b6 100644
--- a/desktop_env/envs/desktop_env.py
+++ b/desktop_env/envs/desktop_env.py
@@ -285,7 +285,7 @@ class DesktopEnv(gym.Env):
         observation = {
             "screenshot": self._get_obs(),
             "accessibility_tree": self.controller.get_accessibility_tree(),
-            "terminal": self.controller.get_terminal_output(),
+            # "terminal": self.controller.get_terminal_output(),
             "instruction": self.instruction
         }
 
diff --git a/lib_run_single.py b/lib_run_single.py
index 82b2dd3..daa374e 100644
--- a/lib_run_single.py
+++ b/lib_run_single.py
@@ -42,7 +42,7 @@ def run_single_example(agent, env, example, max_steps, instruction, args, exampl
                     screenshot = __f.read()
                 _f.write(screenshot)
             # get a11tree and save to wandb
-            thisrun_a11tree = env.controller.get_accessibility_tree()
+            # thisrun_a11tree = env.controller.get_accessibility_tree()
             # str_table.add_data(wandb.Image(data_or_path=os.path.join(example_result_dir, f"step_{step_idx + 1}_{action_timestamp}.png"), caption=f"step_{step_idx + 1}_{action_timestamp}"),
             #                 thisrun_a11tree,
             #                 response, action, action_timestamp, done)