From 014090a8084eec5d3487825612f4832cdda3305c Mon Sep 17 00:00:00 2001
From: mgreminger <michael.greminger@gmail.com>
Date: Mon, 6 Jan 2025 16:03:03 -0600
Subject: [PATCH] fix: fix code generation regression

---
 src/parser/LatexToSympy.ts | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/parser/LatexToSympy.ts b/src/parser/LatexToSympy.ts
index 9391203c..c4cc3a7a 100644
--- a/src/parser/LatexToSympy.ts
+++ b/src/parser/LatexToSympy.ts
@@ -679,11 +679,13 @@ export class LatexToSympy extends LatexParserVisitor<string | Statement | UnitBl
       }
       if (isCodeFunctionQuery) {
 
+        const codeFunctionName = codeFunction.sympy.match(/,(\S*)\)/)[1];
+
         const codeFunctionRawQuery: CodeFunctionRawQuery = {
           type: "query",
           unitlessSubExpressions: [],
           implicitParams: [],
-          params: [codeFunction.sympy,],
+          params: [codeFunctionName,],
           functions: [],
           arguments: [],
           localSubs: [],
@@ -697,7 +699,7 @@ export class LatexToSympy extends LatexParserVisitor<string | Statement | UnitBl
           isScatterXValuesQueryStatement: false,
           isScatterYValuesQueryStatement: false,
           isFromPlotCell: false,
-          sympy: codeFunction.sympy,
+          sympy: codeFunctionName,
           isRange: false,
           isDataTableQuery: false,
           isCodeFunctionQuery: false,
@@ -709,7 +711,7 @@ export class LatexToSympy extends LatexParserVisitor<string | Statement | UnitBl
           isCodeFunctionQuery: true,
           codeFunctionRawQuery: codeFunctionRawQuery,
           generateCode: false,
-          functionName: codeFunction.sympy,
+          functionName: codeFunctionName,
           parameterNames: codeFunction.functionParameters,
           parameterValues: parameterValues,
           parameterUnits: parameterUnits,