diff --git a/server/src/assessment/assessment.module.ts b/server/src/assessment/assessment.module.ts
index 833846b..b0a0f53 100644
--- a/server/src/assessment/assessment.module.ts
+++ b/server/src/assessment/assessment.module.ts
@@ -1,5 +1,7 @@
 import { Module, forwardRef } from '@nestjs/common';
+import { ModuleRef } from '@nestjs/core';
 import { TypeOrmModule } from '@nestjs/typeorm';
+import { DataSource } from 'typeorm';
 import { AssessmentService } from './assessment.service';
 import { AssessmentController } from './assessment.controller';
 import { AssessmentSession } from './entities/assessment-session.entity';
@@ -57,6 +59,12 @@ import { AuditLogService } from './services/audit-log.service';
     QuestionBankService,
     ExportService,
     AuditLogService,
+    {
+      provide: DataSource,
+      useFactory: (moduleRef: ModuleRef) =>
+        moduleRef.get(DataSource, { strict: false }),
+      inject: [ModuleRef],
+    },
   ],
   exports: [AssessmentService, TemplateService, QuestionOutlineService, QuestionBankService, ExportService],
 })