diff --git a/apps/meteor/ee/tests/unit/apps/livechat-enterprise/lib/restrictQuery.tests.ts b/apps/meteor/ee/tests/unit/apps/livechat-enterprise/lib/restrictQuery.tests.ts
index e77a2b0380bdd..5dc3adb558fc6 100644
--- a/apps/meteor/ee/tests/unit/apps/livechat-enterprise/lib/restrictQuery.tests.ts
+++ b/apps/meteor/ee/tests/unit/apps/livechat-enterprise/lib/restrictQuery.tests.ts
@@ -39,7 +39,7 @@ describe('restrictQuery', () => {
 			'@rocket.chat/models': {
 				LivechatDepartment: { find: findStub },
 			},
-			'../methods/getUnitsFromUserRoles': {
+			'@rocket.chat/omni-core-ee': {
 				getUnitsFromUser: getUnitsFromUserStub,
 			},
 			'./logger': {