diff --git a/src/Models/Repositories/CharacterRepository.php b/src/Models/Repositories/CharacterRepository.php
index 7f1cc77..0a05a57 100644
--- a/src/Models/Repositories/CharacterRepository.php
+++ b/src/Models/Repositories/CharacterRepository.php
@@ -45,7 +45,7 @@ class CharacterRepository extends EntityRepository
     /**
      * Find a character by ID.
      */
-    public function find($id, int $deletes = self::SKIP_SOFTDELETED)
+    public function find($id, $lockMode = NULL, $lockVersion = NULL, int $deletes = self::SKIP_SOFTDELETED)
     {
         $queryBuilder = $this->getEntityManager()->createQueryBuilder();
         $queryBuilder->select("c")
diff --git a/src/Models/Repositories/MessageThreadRepository.php b/src/Models/Repositories/MessageThreadRepository.php
index 795eb88..5965977 100644
--- a/src/Models/Repositories/MessageThreadRepository.php
+++ b/src/Models/Repositories/MessageThreadRepository.php
@@ -4,6 +4,7 @@ declare(strict_types = 1);
 
 namespace LotGD\Core\Models\Repositories;
 
+use Doctrine\Common\Util\Debug;
 use Doctrine\ORM\EntityRepository;
 use Doctrine\ORM\NoResultException;
 use Doctrine\ORM\QueryBuilder;
diff --git a/tests/Models/MessageModelTest.php b/tests/Models/MessageModelTest.php
index 8a39af7..5651c87 100644
--- a/tests/Models/MessageModelTest.php
+++ b/tests/Models/MessageModelTest.php
@@ -64,7 +64,7 @@ class MessageModelTest extends CoreModelTestCase
 
         $character1 = $em->getRepository(Character::class)->find(1);
         $character2 = $em->getRepository(Character::class)->find(2);
-        $character3 = $em->getRepository(Character::class)->find(3, CharacterRepository::INCLUDE_SOFTDELETED);
+        $character3 = $em->getRepository(Character::class)->find(3, NULL, NULL, CharacterRepository::INCLUDE_SOFTDELETED);
         $character4 = $em->getRepository(Character::class)->find(4);
         
         $thread1 = $em->getRepository(MessageThread::class)->findOrCreateFor([$character1, $character2, $character3, $character4]);
@@ -92,7 +92,7 @@ class MessageModelTest extends CoreModelTestCase
         
         $character1 = $em->getRepository(Character::class)->find(1);
         $character2 = $em->getRepository(Character::class)->find(2);
-        $character3 = $em->getRepository(Character::class)->find(3, CharacterRepository::INCLUDE_SOFTDELETED);
+        $character3 = $em->getRepository(Character::class)->find(3, NULL, NULL, CharacterRepository::INCLUDE_SOFTDELETED);
         $character4 = $em->getRepository(Character::class)->find(4);
         
         $thread1 = $em->getRepository(MessageThread::class)->findOrCreateFor([$character1, $character2, $character3, $character4]);