diff --git a/tinyflow-java-core/src/main/java/dev/tinyflow/core/parser/impl/LlmNodeParser.java b/tinyflow-java-core/src/main/java/dev/tinyflow/core/parser/impl/LlmNodeParser.java
index f4db44e84ca317249dd41788e02038428c424a21..b005b9c289c1235372ac85f91b7b0e57411370cc 100644
--- a/tinyflow-java-core/src/main/java/dev/tinyflow/core/parser/impl/LlmNodeParser.java
+++ b/tinyflow-java-core/src/main/java/dev/tinyflow/core/parser/impl/LlmNodeParser.java
@@ -44,7 +44,7 @@ public class LlmNodeParser extends BaseNodeParser {
 
         LlmProvider llmProvider = tinyflow.getLlmProvider();
         if (llmProvider != null) {
-            Llm llm = llmProvider.getLlm(data.getBigInteger("llmId"));
+            Llm llm = llmProvider.getLlm(data.getString("llmId"));
             llmNode.setLlm(llm);
         }