diff --git a/src/test/java/org/opensearch/flowframework/rest/FlowFrameworkRestApiIT.java b/src/test/java/org/opensearch/flowframework/rest/FlowFrameworkRestApiIT.java
index 3682487a..a2893172 100644
--- a/src/test/java/org/opensearch/flowframework/rest/FlowFrameworkRestApiIT.java
+++ b/src/test/java/org/opensearch/flowframework/rest/FlowFrameworkRestApiIT.java
@@ -365,7 +365,7 @@ public void testCreateAndProvisionConnectorToolAgentFrameworkWorkflow() throws E
         @SuppressWarnings("unchecked")
         ArrayList<Map<String, Object>> tools = (ArrayList<Map<String, Object>>) agentResponse.get("tools");
         assertEquals(1, tools.size());
-        Map<String, Object> tool = tools.getFirst();
+        Map<String, Object> tool = tools.get(0);
         assertTrue(tool.containsKey("parameters"));
         @SuppressWarnings("unchecked")
         Map<String, String> toolParameters = (Map<String, String>) tool.get("parameters");