diff --git a/openhands/controller/agent_controller.py b/openhands/controller/agent_controller.py
index 7189371f0b..e5a0b24f96 100644
--- a/openhands/controller/agent_controller.py
+++ b/openhands/controller/agent_controller.py
@@ -77,6 +77,7 @@ class AgentController:
         NullObservation,
         ChangeAgentStateAction,
         AgentStateChangedObservation,
+        AgentCondensationObservation,
     )
 
     def __init__(