diff --git a/app/sprinkles/account/tests/Integration/FindUniqueTest.php b/app/sprinkles/account/tests/Integration/FindUniqueTest.php
index cb92f3aa0..79de1a255 100644
--- a/app/sprinkles/account/tests/Integration/FindUniqueTest.php
+++ b/app/sprinkles/account/tests/Integration/FindUniqueTest.php
@@ -37,6 +37,9 @@ public function setUp(): void
         // Setup test database
         $this->setupTestDatabase();
         $this->refreshDatabase();
+
+        $this->ci->classMapper->setClassMapping('user', 'UserFrosting\Sprinkle\Account\Database\Models\User');
+        $this->ci->classMapper->setClassMapping('group', 'UserFrosting\Sprinkle\Account\Database\Models\Group');
     }
 
     /**