diff --git a/packages/services/hmi-server/src/main/java/software/uncharted/terarium/hmiserver/proxies/knowledgemiddleware/KnowledgeMiddlewareProxy.java b/packages/services/hmi-server/src/main/java/software/uncharted/terarium/hmiserver/proxies/knowledgemiddleware/KnowledgeMiddlewareProxy.java
index 92d0fb08f7..07316ed532 100644
--- a/packages/services/hmi-server/src/main/java/software/uncharted/terarium/hmiserver/proxies/knowledgemiddleware/KnowledgeMiddlewareProxy.java
+++ b/packages/services/hmi-server/src/main/java/software/uncharted/terarium/hmiserver/proxies/knowledgemiddleware/KnowledgeMiddlewareProxy.java
@@ -135,9 +135,9 @@ ExtractionResponse postCodeToAMR(
 	/**
 	 * Transform LaTeX equations to AMR
 	 * @param 	equationType (String): [latex, mathml]
-	 * @param 	model (String): AMR model return type. Defaults to "petrinet". Options: "regnet", "petrinet".
+	 * @param 	framework (String): AMR model return type. Defaults to "petrinet". Options: "regnet", "petrinet".
 	 * @param 	modelId (String): the id of the model (to update) based on the set of equations
-	 * @param 	payload (List<String>): the list of LaTeX strings representing the functions that are used to convert to AMR
+	 * @param 	equations (List<String>): the list of LaTeX strings representing the functions that are used to convert to AMR
 	 * @return  (ExtractionResponse)
 	 */
 	@POST