diff --git a/experiments/agents/agent.py b/experiments/agents/agent.py
index 57f3b06..5d3bbe7 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