diff --git a/app/proprietary/src/test/java/stirling/software/proprietary/policy/overview/PolicyOverviewServiceTest.java b/app/proprietary/src/test/java/stirling/software/proprietary/policy/overview/PolicyOverviewServiceTest.java
index 221cbc4482..3224ec05b7 100644
--- a/app/proprietary/src/test/java/stirling/software/proprietary/policy/overview/PolicyOverviewServiceTest.java
+++ b/app/proprietary/src/test/java/stirling/software/proprietary/policy/overview/PolicyOverviewServiceTest.java
@@ -119,7 +119,7 @@ class PolicyOverviewServiceTest {
         properties.getSecurity().setEnableLogin(true);
         UserServiceInterface userService = mock(UserServiceInterface.class);
         PolicyManagementAuthority authority = mock(PolicyManagementAuthority.class);
-        when(authority.currentUserTeamId()).thenReturn(1L);
+        when(authority.requireCurrentUserTeamId()).thenReturn(1L);
         SourceAccessGuard sourceGuard = new SourceAccessGuard(userService, properties, authority);
         PolicyAccessGuard policyGuard = new PolicyAccessGuard(userService, properties, authority);
         PolicyOverviewService scoped =
diff --git a/app/proprietary/src/test/java/stirling/software/proprietary/policy/source/SourceOverviewServiceTest.java b/app/proprietary/src/test/java/stirling/software/proprietary/policy/source/SourceOverviewServiceTest.java
index 275f076db3..0853bf8054 100644
--- a/app/proprietary/src/test/java/stirling/software/proprietary/policy/source/SourceOverviewServiceTest.java
+++ b/app/proprietary/src/test/java/stirling/software/proprietary/policy/source/SourceOverviewServiceTest.java
@@ -109,7 +109,7 @@ class SourceOverviewServiceTest {
         properties.getSecurity().setEnableLogin(true);
         UserServiceInterface userService = mock(UserServiceInterface.class);
         PolicyManagementAuthority authority = mock(PolicyManagementAuthority.class);
-        when(authority.currentUserTeamId()).thenReturn(1L);
+        when(authority.requireCurrentUserTeamId()).thenReturn(1L);
         SourceAccessGuard sourceGuard = new SourceAccessGuard(userService, properties, authority);
         PolicyAccessGuard policyGuard = new PolicyAccessGuard(userService, properties, authority);
         SourceOverviewService scoped =