diff --git a/src/HeartTrackAPI/Program.cs b/src/HeartTrackAPI/Program.cs
index b0010a6..6cd94a6 100644
--- a/src/HeartTrackAPI/Program.cs
+++ b/src/HeartTrackAPI/Program.cs
@@ -1,4 +1,6 @@
+using DbContextLib;
 using HeartTrackAPI.Utils;
+using StubbedContextLib;
 
 var builder = WebApplication.CreateBuilder(args);
 
@@ -16,4 +18,8 @@ var app = builder.Build();
 
 init.Configure(app, app.Environment);
 
+var context = app.Services.GetService<HeartTrackContext>() ?? app.Services.GetService<TrainingStubbedContext>();
+context!.Database.EnsureCreated();
+
+
 app.Run();
\ No newline at end of file