diff --git a/examples/stanford_town/stanford_town.py b/examples/stanford_town/stanford_town.py
index 7c39cbf04..17364f7c8 100644
--- a/examples/stanford_town/stanford_town.py
+++ b/examples/stanford_town/stanford_town.py
@@ -7,7 +7,7 @@ from typing import Any, Optional
 from examples.stanford_town.roles.st_role import STRole
 from examples.stanford_town.utils.const import MAZE_ASSET_PATH
 from metagpt.context import Context
-from metagpt.environment.stanford_town.stanford_town_env import StanfordTownEnv
+from metagpt.environment import StanfordTownEnv
 from metagpt.logs import logger
 from metagpt.team import Team
 
diff --git a/examples/stanford_town/tests/actions/test_gen_action_details.py b/examples/stanford_town/tests/actions/test_gen_action_details.py
index bf6b75c04..af4d94c4d 100644
--- a/examples/stanford_town/tests/actions/test_gen_action_details.py
+++ b/examples/stanford_town/tests/actions/test_gen_action_details.py
@@ -13,8 +13,8 @@ from examples.stanford_town.actions.gen_action_details import (
 )
 from examples.stanford_town.roles.st_role import STRole
 from examples.stanford_town.utils.const import MAZE_ASSET_PATH
+from metagpt.environment import StanfordTownEnv
 from metagpt.environment.api.env_api import EnvAPIAbstract
-from metagpt.environment.stanford_town.stanford_town_env import StanfordTownEnv
 
 
 @pytest.mark.asyncio
diff --git a/examples/stanford_town/tests/plan/test_conversation.py b/examples/stanford_town/tests/plan/test_conversation.py
index 86c2f0b94..ff8664538 100644
--- a/examples/stanford_town/tests/plan/test_conversation.py
+++ b/examples/stanford_town/tests/plan/test_conversation.py
@@ -11,7 +11,7 @@ from examples.stanford_town.roles.st_role import STRole
 from examples.stanford_town.utils.const import MAZE_ASSET_PATH, STORAGE_PATH
 from examples.stanford_town.utils.mg_ga_transform import get_reverie_meta
 from examples.stanford_town.utils.utils import copy_folder
-from metagpt.environment.stanford_town.stanford_town_env import StanfordTownEnv
+from metagpt.environment import StanfordTownEnv
 
 
 async def init_two_roles(fork_sim_code: str = "base_the_ville_isabella_maria_klaus") -> Tuple["STRole"]:
diff --git a/examples/stanford_town/tests/roles/test_st_role.py b/examples/stanford_town/tests/roles/test_st_role.py
index b8feecb2a..b2e4c6969 100644
--- a/examples/stanford_town/tests/roles/test_st_role.py
+++ b/examples/stanford_town/tests/roles/test_st_role.py
@@ -7,7 +7,7 @@ import pytest
 from examples.stanford_town.memory.agent_memory import BasicMemory
 from examples.stanford_town.roles.st_role import STRole
 from examples.stanford_town.utils.const import MAZE_ASSET_PATH
-from metagpt.environment.stanford_town.stanford_town_env import StanfordTownEnv
+from metagpt.environment import StanfordTownEnv
 
 
 @pytest.mark.asyncio
diff --git a/examples/stanford_town/tests/test_reflect.py b/examples/stanford_town/tests/test_reflect.py
index 48422d073..dbcbd974f 100644
--- a/examples/stanford_town/tests/test_reflect.py
+++ b/examples/stanford_town/tests/test_reflect.py
@@ -11,7 +11,7 @@ from examples.stanford_town.actions.run_reflect_action import (
 )
 from examples.stanford_town.roles.st_role import STRole
 from examples.stanford_town.utils.const import MAZE_ASSET_PATH
-from metagpt.environment.stanford_town.stanford_town_env import StanfordTownEnv
+from metagpt.environment import StanfordTownEnv
 
 
 @pytest.mark.asyncio