diff --git a/src/HeartTrackAPI/Dockerfile b/src/HeartTrackAPI/Dockerfile
index a5ea045..560ddd2 100644
--- a/src/HeartTrackAPI/Dockerfile
+++ b/src/HeartTrackAPI/Dockerfile
@@ -30,6 +30,12 @@ FROM build AS publish
 ARG BUILD_CONFIGURATION=Release
 RUN dotnet publish "HeartTrackAPI.csproj" -c $BUILD_CONFIGURATION -o /app/publish /p:UseAppHost=false
 
+# Add the migrations
+RUN dotnet ef migrations add MyMigrations --project StubbedContextLib/ --context TrainingStubbedContext
+
+# Update the database
+RUN dotnet ef database update --project StubbedContextLib/ --context TrainingStubbedContext --startup-project HeartTrackAPI/
+
 FROM base AS final
 WORKDIR /app
 COPY --from=publish /app/publish .