diff --git a/src/main/java/org/opensearch/ubi/UbiActionFilter.java b/src/main/java/org/opensearch/ubi/UbiActionFilter.java
index f01fbf4..64a5887 100644
--- a/src/main/java/org/opensearch/ubi/UbiActionFilter.java
+++ b/src/main/java/org/opensearch/ubi/UbiActionFilter.java
@@ -23,6 +23,7 @@
 import org.opensearch.action.search.SearchResponse;
 import org.opensearch.action.support.ActionFilter;
 import org.opensearch.action.support.ActionFilterChain;
+import org.opensearch.action.support.ActionRequestMetadata;
 import org.opensearch.common.xcontent.XContentType;
 import org.opensearch.core.action.ActionListener;
 import org.opensearch.core.action.ActionResponse;
@@ -79,6 +80,7 @@ public <Request extends ActionRequest, Response extends ActionResponse> void app
         Task task,
         String action,
         Request request,
+        ActionRequestMetadata<Request, Response> actionRequestMetadata,
         ActionListener<Response> listener,
         ActionFilterChain<Request, Response> chain
     ) {
diff --git a/src/test/java/org/opensearch/ubi/UbiActionFilterTests.java b/src/test/java/org/opensearch/ubi/UbiActionFilterTests.java
index c7a9a03..f8c9203 100644
--- a/src/test/java/org/opensearch/ubi/UbiActionFilterTests.java
+++ b/src/test/java/org/opensearch/ubi/UbiActionFilterTests.java
@@ -14,6 +14,7 @@
 import org.opensearch.action.search.SearchRequest;
 import org.opensearch.action.search.SearchResponse;
 import org.opensearch.action.support.ActionFilterChain;
+import org.opensearch.action.support.ActionRequestMetadata;
 import org.opensearch.common.action.ActionFuture;
 import org.opensearch.common.settings.Settings;
 import org.opensearch.common.unit.TimeValue;
@@ -98,7 +99,7 @@ public void testApplyWithoutUbiBlock() {
 
         when(request.source()).thenReturn(searchSourceBuilder);
 
-        ubiActionFilter.apply(task, "ubi", request, listener, chain);
+        ubiActionFilter.apply(task, "ubi", request, ActionRequestMetadata.empty(), listener, chain);
 
         verify(client, never()).index(any(), any());
 
@@ -157,7 +158,7 @@ public void testApplyWithUbiBlockWithoutQueryId() {
 
         when(request.source()).thenReturn(searchSourceBuilder);
 
-        ubiActionFilter.apply(task, "ubi", request, listener, chain);
+        ubiActionFilter.apply(task, "ubi", request, ActionRequestMetadata.empty(), listener, chain);
 
         verify(client, atMostOnce()).index(any(), any());