diff --git a/test/framework/src/main/java/org/opensearch/test/rest/RestActionTestCase.java b/test/framework/src/main/java/org/opensearch/test/rest/RestActionTestCase.java
index 7d1e393a82bb7..eb3cb98fc113a 100644
--- a/test/framework/src/main/java/org/opensearch/test/rest/RestActionTestCase.java
+++ b/test/framework/src/main/java/org/opensearch/test/rest/RestActionTestCase.java
@@ -122,7 +122,7 @@ public void reset() {
          * @param verifier A function which is called in place of {@link #doExecute(ActionType, ActionRequest, ActionListener)}
          */
         public <Request extends ActionRequest, Response extends ActionResponse> void setExecuteVerifier(
-            BiFunction<ActionType<Response>, Request, Void> verifier
+            BiFunction<ActionType<Response>, Request, Response> verifier
         ) {
             executeVerifier.set(verifier);
         }
@@ -142,7 +142,7 @@ public <Request extends ActionRequest, Response extends ActionResponse> void doE
          * @param verifier A function which is called in place of {@link #executeLocally(ActionType, ActionRequest, TaskListener)}
          */
         public <Request extends ActionRequest, Response extends ActionResponse> void setExecuteLocallyVerifier(
-            BiFunction<ActionType<Response>, Request, Void> verifier
+            BiFunction<ActionType<Response>, Request, Response> verifier
         ) {
             executeLocallyVerifier.set(verifier);
         }