diff --git a/gateway-ha/src/test/java/io/trino/gateway/ha/security/TestLbAuthorizer.java b/gateway-ha/src/test/java/io/trino/gateway/ha/security/TestLbAuthorizer.java
index c373931e1..79ec5345c 100644
--- a/gateway-ha/src/test/java/io/trino/gateway/ha/security/TestLbAuthorizer.java
+++ b/gateway-ha/src/test/java/io/trino/gateway/ha/security/TestLbAuthorizer.java
@@ -1,5 +1,8 @@
 package io.trino.gateway.ha.security;
 
+import static org.junit.jupiter.api.Assertions.assertFalse;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+
 import io.trino.gateway.ha.config.AuthorizationConfiguration;
 import lombok.extern.slf4j.Slf4j;
 import org.junit.jupiter.api.BeforeAll;
@@ -8,9 +11,6 @@
 import java.util.Optional;
 import java.util.regex.PatternSyntaxException;
 
-import static org.junit.Assert.assertFalse;
-import static org.junit.Assert.assertTrue;
-
 @Slf4j
 public class TestLbAuthorizer {
     private static final String USER = "username";
@@ -46,11 +46,11 @@ static void configureRole(String regex, String role) {
     }
 
     static void assertMatch(String role) {
-        assertTrue(authorizer.authorize(principal, role));
+        assertTrue(authorizer.authorize(principal, role, null));
     }
 
     static void assertNotMatch(String role) {
-        assertFalse(authorizer.authorize(principal, role));
+        assertFalse(authorizer.authorize(principal, role, null));
     }
 
     static void assertBadPattern(String role) {