2 files changed, 3 insertions, 3 deletions
diff --git a/Timeline.Tests/Helpers/TestApplication.cs b/Timeline.Tests/Helpers/TestApplication.cs
index a624da6b..d18f2848 100644
--- a/Timeline.Tests/Helpers/TestApplication.cs
+++ b/Timeline.Tests/Helpers/TestApplication.cs
@@ -18,7 +18,7 @@ namespace Timeline.Tests.Helpers
                 builder.ConfigureServices(services =>
                 {
                     services.AddEntityFrameworkSqlite();
-                    services.AddDbContext<DatabaseContext>(options =>
+                    services.AddDbContext<DatabaseContext, DevelopmentDatabaseContext>(options =>
                     {
                         options.UseSqlite(Database.Connection);
                     });
diff --git a/Timeline.Tests/Helpers/TestDatabase.cs b/Timeline.Tests/Helpers/TestDatabase.cs
index 10224c27..9560f353 100644
--- a/Timeline.Tests/Helpers/TestDatabase.cs
+++ b/Timeline.Tests/Helpers/TestDatabase.cs
@@ -47,11 +47,11 @@ namespace Timeline.Tests.Helpers
             Connection = new SqliteConnection("Data Source=:memory:;");
             Connection.Open();
 
-            var options = new DbContextOptionsBuilder<DatabaseContext>()
+            var options = new DbContextOptionsBuilder<DevelopmentDatabaseContext>()
                 .UseSqlite(Connection)
                 .Options;
 
-            Context = new DatabaseContext(options);
+            Context = new DevelopmentDatabaseContext(options);
 
             InitDatabase(Context);
         }