diff --git a/src/HeartTrackAPI/Dockerfile b/src/HeartTrackAPI/Dockerfile
index fac2085..8656c8e 100644
--- a/src/HeartTrackAPI/Dockerfile
+++ b/src/HeartTrackAPI/Dockerfile
@@ -21,15 +21,9 @@ COPY ["StubAPI/StubAPI.csproj", "StubAPI/"]
 COPY ["StubbedContextLib/StubbedContextLib.csproj", "StubbedContextLib/"]
 RUN dotnet restore "HeartTrackAPI/HeartTrackAPI.csproj"
 COPY . .
-RUN dotnet tool install --global dotnet-ef --version 8.0
 
-ENV PATH="${PATH}:/root/.dotnet/tools"
-# Add the migrations
-RUN dotnet-ef migrations add --project StubbedContextLib/StubbedContextLib.csproj --startup-project HeartTrackAPI/HeartTrackAPI.csproj --context StubbedContextLib.TrainingStubbedContext --configuration Debug Initial 
-# Update the database
-RUN dotnet-ef database update --project StubbedContextLib/StubbedContextLib.csproj --startup-project HeartTrackAPI/HeartTrackAPI.csproj --context StubbedContextLib.TrainingStubbedContext --configuration Debug 
 WORKDIR "/src/HeartTrackAPI"
-RUN ls
+
 RUN dotnet build "HeartTrackAPI.csproj" -c $BUILD_CONFIGURATION -o /app/build
 
 FROM build AS publish
@@ -38,6 +32,13 @@ RUN dotnet publish "HeartTrackAPI.csproj" -c $BUILD_CONFIGURATION -o /app/publis
 FROM base AS final
 WORKDIR /app
 COPY --from=publish /app/publish .
-COPY --from=publish /src/HeartTrackAPI/uca.HeartTrack.db .
-RUN chown $APP_UID:$APP_UID uca.HeartTrack.db
+RUN dotnet tool install --global dotnet-ef --version 8.0
+
+ENV PATH="${PATH}:/root/.dotnet/tools"
+# Add the migrations
+RUN dotnet-ef migrations add --project StubbedContextLib/StubbedContextLib.csproj --startup-project HeartTrackAPI/HeartTrackAPI.csproj --context StubbedContextLib.TrainingStubbedContext --configuration Debug Initial 
+# Update the database
+RUN dotnet-ef database update --project StubbedContextLib/StubbedContextLib.csproj --startup-project HeartTrackAPI/HeartTrackAPI.csproj --context StubbedContextLib.TrainingStubbedContext --configuration Debug 
+
+# COPY --from=publish /src/HeartTrackAPI/uca.HeartTrack.db .
 ENTRYPOINT ["dotnet", "HeartTrackAPI.dll"]
\ No newline at end of file