diff --git a/packages/nocodb/src/db/BaseModelSqlv2.ts b/packages/nocodb/src/db/BaseModelSqlv2.ts
index d474f8de09..97844a482f 100644
--- a/packages/nocodb/src/db/BaseModelSqlv2.ts
+++ b/packages/nocodb/src/db/BaseModelSqlv2.ts
@@ -2817,7 +2817,7 @@ class BaseModelSqlv2 {
           ? response?.[0]?.[ai.id]
           : response?.[ai.id];
         response = await this.readByPk(
-          this.extractCompositePK({ rowId: id, insertObj, ag }),
+          this.extractCompositePK({ rowId: id, insertObj, ag, ai }),
           false,
           {},
           { ignoreView: true, getHiddenColumn: true },
@@ -3250,8 +3250,8 @@ class BaseModelSqlv2 {
     insertObj,
     force = false,
   }: {
-    ai?: Column<any>;
-    ag?: Column<any>;
+    ai: Column<any>;
+    ag: Column<any>;
     rowId;
     insertObj: Record<string, any>;
     force?: boolean;