diff --git a/tests/integration/Espo/Core/Formula/FormulaTest.php b/tests/integration/Espo/Core/Formula/FormulaTest.php
index 77e999473c..4e6e8f5ec2 100644
--- a/tests/integration/Espo/Core/Formula/FormulaTest.php
+++ b/tests/integration/Espo/Core/Formula/FormulaTest.php
@@ -840,7 +840,7 @@ class FormulaTest extends BaseTestCase
         $script = "ext\\email\\applyTemplate('{$email->getId()}', '{$emailTemplate->getId()}')";
         $fm->run($script);
 
-        $email = $em->getEntity('Email', $email->getId());
+        $email = $em->getEntityById('Email', $email->getId());
 
         $this->assertEquals('Test', $email->get('name'));
         $this->assertEquals('Test Contact 1 Hello, Case 1', $email->get('body'));
@@ -865,7 +865,7 @@ class FormulaTest extends BaseTestCase
 
         $this->assertIsString($id);
 
-        $attachment = $em->getEntity('Attachment', $id);
+        $attachment = $em->getEntityById('Attachment', $id);
 
         $this->assertNotNull($attachment);
         $this->assertEquals('test.pdf', $attachment->get('name'));
@@ -875,7 +875,7 @@ class FormulaTest extends BaseTestCase
         $script = "ext\\pdf\\generate('Account', '{$a->getId()}', '{$template->getId()}', 'test.pdf')";
         $id = $fm->run($script);
 
-        $attachment = $em->getEntity('Attachment', $id);
+        $attachment = $em->getEntityById('Attachment', $id);
 
         $this->assertEquals('test.pdf', $attachment->get('name'));
 
@@ -883,7 +883,7 @@ class FormulaTest extends BaseTestCase
         $script = "ext\\pdf\\generate('Account', '{$a->getId()}', '{$template->getId()}')";
         $id = $fm->run($script);
 
-        $attachment = $em->getEntity('Attachment', $id);
+        $attachment = $em->getEntityById('Attachment', $id);
 
         $this->assertEquals('1.pdf', $attachment->get('name'));
     }
diff --git a/tests/integration/Espo/Core/Utils/FieldManager/ArrayTypeTest.php b/tests/integration/Espo/Core/Utils/FieldManager/ArrayTypeTest.php
index 164906378f..272f0802cd 100644
--- a/tests/integration/Espo/Core/Utils/FieldManager/ArrayTypeTest.php
+++ b/tests/integration/Espo/Core/Utils/FieldManager/ArrayTypeTest.php
@@ -88,7 +88,7 @@ class ArrayTypeTest extends BaseTestCase
 
         $entityManager = $app->getContainer()->getByClass(EntityManager::class);
 
-        $account = $entityManager->getEntity('Account');
+        $account = $entityManager->getNewEntity('Account');
         $account->set([
             'name' => 'Test',
             'cTestArray' => ['option1', 'option3']
@@ -96,7 +96,7 @@ class ArrayTypeTest extends BaseTestCase
 
         $entityManager->saveEntity($account);
 
-        $account = $entityManager->getEntity('Account', $account->getId());
+        $account = $entityManager->getEntityById('Account', $account->getId());
         $this->assertEquals(['option1', 'option3'], $account->get('cTestArray'));
     }
 
@@ -123,7 +123,7 @@ class ArrayTypeTest extends BaseTestCase
         $this->assertTrue($savedFieldDefs['required']);
 
         $entityManager = $app->getContainer()->getByClass(EntityManager::class);
-        $account = $entityManager->getEntity('Account');
+        $account = $entityManager->getNewEntity('Account');
         $account->set([
             'name' => 'Test',
         ]);
diff --git a/tests/integration/Espo/Core/Utils/FieldManager/EnumTypeTest.php b/tests/integration/Espo/Core/Utils/FieldManager/EnumTypeTest.php
index 7806d68365..787a4c88d2 100644
--- a/tests/integration/Espo/Core/Utils/FieldManager/EnumTypeTest.php
+++ b/tests/integration/Espo/Core/Utils/FieldManager/EnumTypeTest.php
@@ -87,7 +87,7 @@ class EnumTypeTest extends BaseTestCase
 
         $entityManager = $app->getContainer()->getByClass(EntityManager::class);
 
-        $account = $entityManager->getEntity('Account');
+        $account = $entityManager->getNewEntity('Account');
         $account->set([
             'name' => 'Test',
             'cTestEnum' => 'option1',
@@ -95,7 +95,7 @@ class EnumTypeTest extends BaseTestCase
 
         $entityManager->saveEntity($account);
 
-        $account = $entityManager->getEntity('Account', $account->getId());
+        $account = $entityManager->getEntityById('Account', $account->getId());
         $this->assertEquals('option1', $account->get('cTestEnum'));
     }
 
@@ -127,14 +127,14 @@ class EnumTypeTest extends BaseTestCase
 
         $entityManager = $app->getContainer()->getByClass(EntityManager::class);
 
-        $account = $entityManager->getEntity('Account');
+        $account = $entityManager->getNewEntity('Account');
         $account->set([
             'name' => 'New Test',
         ]);
 
         $entityManager->saveEntity($account);
 
-        $account = $entityManager->getEntity('Account', $account->getId());
+        $account = $entityManager->getEntityById('Account', $account->getId());
         $this->assertEquals('option3', $account->get('cTestEnum'));
     }
 }