diff --git a/src/envs/browsergym_env/client.py b/src/envs/browsergym_env/client.py
index e92d967e..2de2f6e7 100644
--- a/src/envs/browsergym_env/client.py
+++ b/src/envs/browsergym_env/client.py
@@ -3,7 +3,7 @@
 from typing import Any, Dict
 
 from openenv_core.http_env_client import HTTPEnvClient, StepResult
-from browsergym_env.models import (
+from .models import (
     BrowserGymAction,
     BrowserGymObservation,
     BrowserGymState,