1 files changed, 8 insertions, 9 deletions
diff --git a/BackEnd/Timeline/Program.cs b/BackEnd/Timeline/Program.cs
index 75bf6154..19fa6e37 100644
--- a/BackEnd/Timeline/Program.cs
+++ b/BackEnd/Timeline/Program.cs
@@ -4,8 +4,10 @@ using Microsoft.Extensions.Configuration;
 using Microsoft.Extensions.DependencyInjection;
 using Microsoft.Extensions.Hosting;
 using System.Resources;
+using System.Threading.Tasks;
 using Timeline.Entities;
 using Timeline.Services;
+using Timeline.Services.Migration;
 
 [assembly: NeutralResourcesLanguage("en")]
 
@@ -13,22 +15,19 @@ namespace Timeline
 {
     public static class Program
     {
-        public static void Main(string[] args)
+        public async static Task Main(string[] args)
         {
             var host = CreateWebHostBuilder(args).Build();
 
-            var env = host.Services.GetRequiredService<IWebHostEnvironment>();
-
             var databaseBackupService = host.Services.GetRequiredService<IDatabaseBackupService>();
             databaseBackupService.BackupNow();
 
-            if (env.IsProduction())
+            using (var scope = host.Services.CreateScope())
             {
-                using (var scope = host.Services.CreateScope())
-                {
-                    var databaseContext = scope.ServiceProvider.GetRequiredService<DatabaseContext>();
-                    databaseContext.Database.Migrate();
-                }
+                var databaseContext = scope.ServiceProvider.GetRequiredService<DatabaseContext>();
+                await databaseContext.Database.MigrateAsync();
+                var customMigrationManager = scope.ServiceProvider.GetRequiredService<ICustomMigrationManager>();
+                await customMigrationManager.Migrate();
             }
 
             host.Run();