diff --git a/src/agentlab/ui_assistant.py b/src/agentlab/ui_assistant.py
index 2bebaa412..96bbb0f9c 100644
--- a/src/agentlab/ui_assistant.py
+++ b/src/agentlab/ui_assistant.py
@@ -3,6 +3,7 @@
 from browsergym.experiments.loop import EnvArgs, ExpArgs
 
 from agentlab.agents.agent_args import AgentArgs
+from agentlab.agents.generic_agent.generic_agent import GenericAgentArgs
 from agentlab.experiments.exp_utils import RESULTS_DIR
 from agentlab.experiments.launch_exp import import_object
 
@@ -14,6 +15,9 @@ def make_exp_args(agent_args: AgentArgs, start_url="https://www.google.com"):
     except AttributeError:
         pass
 
+    if isinstance(agent_args, GenericAgentArgs):
+        agent_args.flags.enable_chat = True
+
     exp_args = ExpArgs(
         agent_args=agent_args,
         env_args=EnvArgs(