diff --git a/backend/src/services/project/project-dal.ts b/backend/src/services/project/project-dal.ts
index f18542216..8b2971595 100644
--- a/backend/src/services/project/project-dal.ts
+++ b/backend/src/services/project/project-dal.ts
@@ -80,13 +80,13 @@ export const projectDALFactory = (db: TDbClient) => {
     }
   };
 
-  const setProjectUpgradeStatus = async (projectId: string, status: ProjectUpgradeStatus) => {
+  const setProjectUpgradeStatus = async (projectId: string, status: ProjectUpgradeStatus | null, tx?: Knex) => {
     try {
       const data: TProjectsUpdate = {
         upgradeStatus: status
       } as const;
 
-      await db(TableName.Project).where({ id: projectId }).update(data);
+      await (tx || db)(TableName.Project).where({ id: projectId }).update(data);
     } catch (error) {
       throw new DatabaseError({ error, name: "Set project upgrade status" });
     }