diff --git a/src/HeartTrackAPI/Dockerfile b/src/HeartTrackAPI/Dockerfile
index ca1c4ec..6514b7e 100644
--- a/src/HeartTrackAPI/Dockerfile
+++ b/src/HeartTrackAPI/Dockerfile
@@ -24,10 +24,9 @@ COPY . .
 RUN ls 
 
 # Add the migrations
-RUN dotnet ef migrations add MyMigrations --project StubbedContextLib/ --context TrainingStubbedContext
-
+RUN dotnet ef migrations add --project StubbedContextLib/ --startup-project HeartTrackAPI/ --context StubbedContextLib.TrainingStubbedContext --configuration Debug Initial --output-dir Migrations
 # Update the database
-RUN dotnet ef database update --project StubbedContextLib/ --context TrainingStubbedContext --startup-project HeartTrackAPI/
+RUN dotnet ef database update --project StubbedContextLib/ --startup-project HeartTrackAPI/ --context StubbedContextLib.TrainingStubbedContext --configuration Debug 
 
 WORKDIR "/src/HeartTrackAPI"
 RUN ls