diff --git a/packages/nocodb/tests/unit/rest/tests/column.test.ts b/packages/nocodb/tests/unit/rest/tests/column.test.ts
index 5c83930922..4afa68eb97 100644
--- a/packages/nocodb/tests/unit/rest/tests/column.test.ts
+++ b/packages/nocodb/tests/unit/rest/tests/column.test.ts
@@ -52,8 +52,8 @@ function columnTests() {
         ],
       });
 
-      const pkColumn = (await country.getColumns()).find((column) => column.pk);
-      const fkColumn = (await city.getColumns()).find(
+      const pkColumn = (await country.getColumns(context)).find((column) => column.pk);
+      const fkColumn = (await city.getColumns(context)).find(
         (column) => column.title === 'CountryId',
       );