diff --git a/pipelines/tools/Dockerfile.finetune b/pipelines/tools/Dockerfile.finetune
index 7be9a78..c33d4a1 100644
--- a/pipelines/tools/Dockerfile.finetune
+++ b/pipelines/tools/Dockerfile.finetune
@@ -1,7 +1,7 @@
 # Unsloth fine-tuning container for Qwen 3.5 4B on RTX 3090.
 #
 # Build:
-#   docker build -f python/prompt_bench/Dockerfile.finetune -t bill-finetune .
+#   docker build -f pipelines/pipelines/tools/Dockerfile.finetune -t bill-finetune .
 #
 # Run:
 #   docker run --rm --device=nvidia.com/gpu=all --ipc=host \
@@ -17,10 +17,9 @@ FROM ghcr.io/unslothai/unsloth:latest
 RUN pip install --no-cache-dir typer rouge-score
 
 WORKDIR /workspace
-COPY python/prompt_bench/finetune.py python/prompt_bench/finetune.py
+COPY pipelines/pipelines/tools/__init__.py pipelines/tools/__init__.py
+COPY pipelines/pipelines/tools/finetune.py pipelines/tools/finetune.py
+COPY pipelines/pipelines/tools/summarization_eval.py pipelines/tools/summarization_eval.py
 COPY config/prompts/summarization_prompts.toml config/prompts/summarization_prompts.toml
-COPY python/prompt_bench/__init__.py python/prompt_bench/__init__.py
-COPY python/prompt_bench/summarization_eval.py python/prompt_bench/summarization_eval.py
-COPY python/__init__.py python/__init__.py
 
-ENTRYPOINT ["python", "-m", "pipelines.prompt_bench.finetune"]
+ENTRYPOINT ["python", "-m", "pipelines.tools.finetune"]
diff --git a/pipelines/tools/containers/finetune.py b/pipelines/tools/containers/finetune.py
index 82455cd..a6f81e2 100644
--- a/pipelines/tools/containers/finetune.py
+++ b/pipelines/tools/containers/finetune.py
@@ -27,7 +27,7 @@ def build_image() -> None:
             "docker",
             "build",
             "-f",
-            str(REPO_DIR / "python/prompt_bench/Dockerfile.finetune"),
+            str(REPO_DIR / "pipelines/pipelines/tools/Dockerfile.finetune"),
             "-t",
             FINETUNE_IMAGE,
             ".",