diff --git a/plugin/src/test/java/org/opensearch/ml/rest/RestMLRAGSearchProcessorIT.java b/plugin/src/test/java/org/opensearch/ml/rest/RestMLRAGSearchProcessorIT.java
index e3cf0d5ddf..23b2bfd501 100644
--- a/plugin/src/test/java/org/opensearch/ml/rest/RestMLRAGSearchProcessorIT.java
+++ b/plugin/src/test/java/org/opensearch/ml/rest/RestMLRAGSearchProcessorIT.java
@@ -199,6 +199,9 @@ public class RestMLRAGSearchProcessorIT extends RestMLRemoteInferenceIT {
         + "\",\n"
         + "    \"secret_key\": \""
         + AWS_SECRET_ACCESS_KEY
+        + "\",\n"
+        + "    \"session_token\": \""
+        + AWS_SESSION_TOKEN
         + "\"\n"
         + "  },\n"
         + "  \"actions\": [\n"
@@ -239,6 +242,9 @@ public class RestMLRAGSearchProcessorIT extends RestMLRemoteInferenceIT {
         + "\",\n"
         + "    \"secret_key\": \""
         + AWS_SECRET_ACCESS_KEY
+        + "\",\n"
+        + "    \"session_token\": \""
+        + AWS_SESSION_TOKEN
         + "\"\n"
         + "  },\n"
         + "  \"actions\": [\n"