diff --git a/rock/sdk/builder/swe_rebench.py b/rock/sdk/builder/swe_rebench.py
index 207e1db3..ed6c3367 100644
--- a/rock/sdk/builder/swe_rebench.py
+++ b/rock/sdk/builder/swe_rebench.py
@@ -1,9 +1,9 @@
 from pathlib import Path
 
 from rock.logger import init_logger
-from rock.sdk.envs.builder.base import EnvBuilder
-from rock.sdk.envs.builder.provider.docker import Docker
-from rock.sdk.envs.builder.provider.dockerfile_builder import DockerfileBuilder
+from rock.sdk.builder.base import EnvBuilder
+from rock.sdk.builder.provider.docker import Docker
+from rock.sdk.builder.provider.dockerfile_builder import DockerfileBuilder
 
 logger = init_logger(__name__)
 
diff --git a/rock/sdk/builder/terminal_bench.py b/rock/sdk/builder/terminal_bench.py
index 9fdfef74..45b12f3e 100644
--- a/rock/sdk/builder/terminal_bench.py
+++ b/rock/sdk/builder/terminal_bench.py
@@ -1,9 +1,9 @@
 from pathlib import Path
 
 from rock.logger import init_logger
-from rock.sdk.envs.builder.base import EnvBuilder
-from rock.sdk.envs.builder.provider.docker import Docker
-from rock.sdk.envs.builder.provider.dockerfile_builder import DockerfileBuilder
+from rock.sdk.builder.base import EnvBuilder
+from rock.sdk.builder.provider.docker import Docker
+from rock.sdk.builder.provider.dockerfile_builder import DockerfileBuilder
 
 logger = init_logger(__name__)