diff --git a/src/main/java/org/opensearch/flowframework/util/ParseUtils.java b/src/main/java/org/opensearch/flowframework/util/ParseUtils.java
index a27cbb2e0..97b74ae31 100644
--- a/src/main/java/org/opensearch/flowframework/util/ParseUtils.java
+++ b/src/main/java/org/opensearch/flowframework/util/ParseUtils.java
@@ -474,11 +474,12 @@ public static Map<String, String> convertStringToObjectMapToStringToStringMap(Ma
     }
 
     /**
-     * Checks if the inputs map contains the specified key and parses the associated value to a Boolean.
+     * Checks if the inputs map contains the specified key and parses the associated value to a generic class.
      *
      * @param inputs the map containing the input data
      * @param key the key to check in the map
-     * @return the Boolean value associated with the key if present, or null if the key is not found
+     * @param type the class to parse the value to
+     * @return the generic type value associated with the key if present, or null if the key is not found
      */
     public static <T> T parseIfExists(Map<String, Object> inputs, String key, Class<T> type) {
         if (!inputs.containsKey(key)) {