diff --git a/asi/agent.py b/asi/agent.py
index a34b715..cb87592 100644
--- a/asi/agent.py
+++ b/asi/agent.py
@@ -73,6 +73,9 @@ def __init__(
             raise ValueError(f"Either use_html or use_axtree must be set to True.")
 
         custom_actions = ACTION_DICT["general"] + ACTION_DICT["webarena"]
+        for website in websites:
+            custom_actions.extend(ACTION_DICT[website])
+
         
 
         self.action_set = CustomActionSet(