diff --git a/pipelines/tools/prompt_bench.py b/pipelines/tools/prompt_bench.py
index 60391c5..c126fc5 100644
--- a/pipelines/tools/prompt_bench.py
+++ b/pipelines/tools/prompt_bench.py
@@ -11,8 +11,8 @@ from typing import Annotated
 
 import typer
 
-from pipelines.tools.containers.lib import check_gpu_free
-from pipelines.tools.containers.vllm import start_vllm, stop_vllm
+from pipelines.containers.lib import check_gpu_free
+from pipelines.containers.vllm import start_vllm, stop_vllm
 from pipelines.tools.downloader import is_model_present
 from pipelines.tools.models import BenchmarkConfig
 from pipelines.tools.vllm_client import VLLMClient