diff --git a/modules/ml-pipeline/src/pipeline/.devcontainer/devcontainer.json b/modules/ml-pipeline/src/pipeline/.devcontainer/devcontainer.json
index 222e7be..968a69a 100644
--- a/modules/ml-pipeline/src/pipeline/.devcontainer/devcontainer.json
+++ b/modules/ml-pipeline/src/pipeline/.devcontainer/devcontainer.json
@@ -6,7 +6,7 @@
 		// Sets the run context to one level up instead of the .devcontainer folder.
 		"context": "..",
 		// Update the 'dockerFile' property if you aren't using the standard 'Dockerfile' filename.
-		"dockerfile": "../Dockerfile"
+		"dockerfile": "../Development.Dockerfile"
 	},
 	"features": {
 		"ghcr.io/devcontainers/features/git:1": {}
diff --git a/modules/ml-pipeline/src/pipeline/Development.Dockerfile b/modules/ml-pipeline/src/pipeline/Development.Dockerfile
index 043d516..69809e9 100644
--- a/modules/ml-pipeline/src/pipeline/Development.Dockerfile
+++ b/modules/ml-pipeline/src/pipeline/Development.Dockerfile
@@ -1,8 +1,7 @@
 # Dockerfile that can be used for development
 FROM python:3.10.12-slim
 
-COPY ~/.ssh ~/.ssh
-COPY training/requirements/requirements-dev.txt requirements.txt
+COPY src/requirements/training/requirements-dev.txt requirements.txt
 
 RUN pip install --upgrade pip
 RUN pip install -r requirements.txt