diff --git a/nestjs/src/courses/course-schedule/course-schedule.service.spec.ts b/nestjs/src/courses/course-schedule/course-schedule.service.spec.ts
index 0bd1643d0..2e4191489 100644
--- a/nestjs/src/courses/course-schedule/course-schedule.service.spec.ts
+++ b/nestjs/src/courses/course-schedule/course-schedule.service.spec.ts
@@ -15,6 +15,8 @@ import {
   TaskInterviewResult,
   TaskResult,
   TaskSolution,
+  TeamDistribution,
+  TeamDistributionStudent,
 } from '@entities/index';
 
 const MOCK_CURRENT_TIME = new Date('2022-03-22T00:00:00.000Z');
@@ -111,6 +113,14 @@ describe('CourseScheduleService', () => {
           provide: getRepositoryToken(TaskChecker),
           useValue: {},
         },
+        {
+          provide: getRepositoryToken(TeamDistributionStudent),
+          useValue: {},
+        },
+        {
+          provide: getRepositoryToken(TeamDistribution),
+          useValue: {},
+        },
       ],
     }).compile();