From 0fc83dfffcc628623434c085b049a7ddba0de01f Mon Sep 17 00:00:00 2001
From: Maarten Zuidhoorn <maarten@zuidhoorn.com>
Date: Fri, 4 Nov 2022 15:06:01 +0100
Subject: [PATCH] Fix mock

---
 packages/controllers/src/test-utils/controller.ts | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/packages/controllers/src/test-utils/controller.ts b/packages/controllers/src/test-utils/controller.ts
index 961e51d928..5db1aab665 100644
--- a/packages/controllers/src/test-utils/controller.ts
+++ b/packages/controllers/src/test-utils/controller.ts
@@ -4,6 +4,7 @@ import {
   AllowedActions,
   AllowedEvents,
   CheckSnapBlockListArg,
+  handlerEndowments,
   PersistedSnapControllerState,
   SnapController,
   SnapControllerActions,
@@ -166,7 +167,7 @@ export const getSnapControllerWithEESOptions = (
 
       if (
         method === 'PermissionController:hasPermission' &&
-        args[1] === SnapEndowments.Rpc
+        Object.values(handlerEndowments).includes(args[1])
       ) {
         return true;
       }