diff --git a/WD7UVN_HFT_2023241.Test/DeleteTests.cs b/WD7UVN_HFT_2023241.Test/DeleteTests.cs
index ca0a2c7..3f8d6de 100644
--- a/WD7UVN_HFT_2023241.Test/DeleteTests.cs
+++ b/WD7UVN_HFT_2023241.Test/DeleteTests.cs
@@ -15,6 +15,7 @@ namespace WD7UVN_HFT_2023241.Test
         public void Setup()
         {
             logic = new LogicServices(new CRUD());
+            Database.Context = new CompanyDbContext();
         }
 
         [Test]
diff --git a/WD7UVN_HFT_2023241.Test/UpdateTests.cs b/WD7UVN_HFT_2023241.Test/UpdateTests.cs
index 95babba..d3cca86 100644
--- a/WD7UVN_HFT_2023241.Test/UpdateTests.cs
+++ b/WD7UVN_HFT_2023241.Test/UpdateTests.cs
@@ -13,6 +13,7 @@ namespace WD7UVN_HFT_2023241.Test
         public void Setup()
         {
             logic = new LogicServices(new CRUD());
+            Database.Context = new CompanyDbContext();
         }
 
         [Test]