diff --git a/experiments/agents/agent.py b/experiments/agents/agent.py
index 96b0ae2..c31e6b2 100644
--- a/experiments/agents/agent.py
+++ b/experiments/agents/agent.py
@@ -1,4 +1,4 @@
-from base import Agent as BaseAgent
+from .base import Agent as BaseAgent
 from browser_use import Browser, Agent, ChatOpenAI
 from enum import Enum