diff --git a/packages/nocodb/src/db/formulav2/formulaQueryBuilderv2.ts b/packages/nocodb/src/db/formulav2/formulaQueryBuilderv2.ts
index b9294e6591..993dedf412 100644
--- a/packages/nocodb/src/db/formulav2/formulaQueryBuilderv2.ts
+++ b/packages/nocodb/src/db/formulav2/formulaQueryBuilderv2.ts
@@ -80,9 +80,9 @@ async function _formulaQueryBuilder(
     tree = validateFormulaAndExtractTreeWithType({
       formula: _tree.replaceAll('{{', '{').replaceAll('}}', '}'),
       columns,
-      clientOrSqlUi: await Source.get(column.source_id).then(
-        (source) => source.type,
-      ),
+      clientOrSqlUi: await Source.get(
+        model?.source_id ?? column?.source_id,
+      ).then((source) => source.type),
     });
 
     // populate and save parsedTree to column if not exist