diff --git a/src/pipeline-tasks/pipeline-tasks.service.spec.ts b/src/pipeline-tasks/pipeline-tasks.service.spec.ts
index ddce7df..8f64810 100644
--- a/src/pipeline-tasks/pipeline-tasks.service.spec.ts
+++ b/src/pipeline-tasks/pipeline-tasks.service.spec.ts
@@ -32,6 +32,7 @@ describe('PipelineTasksService', () => {
     ({
       pipelineId: 'test',
       commit: 'test',
+      pipeline: { branch: 'master' },
       units: [],
     } as PipelineTask);
 
@@ -79,6 +80,7 @@ describe('PipelineTasksService', () => {
     jest
       .spyOn(taskRepository, 'create')
       .mockImplementation((data: any) => data);
+    jest.spyOn(taskRepository, 'findOne').mockImplementation(async () => null);
   });
 
   it('should be defined', () => {
@@ -103,6 +105,7 @@ describe('PipelineTasksService', () => {
       const save = jest
         .spyOn(taskRepository, 'save')
         .mockImplementation(async (data: any) => data);
+      const findOne = jest.spyOn(taskRepository, 'findOne');
       jest
         .spyOn(service, 'doNextTask')
         .mockImplementation(async () => undefined);
@@ -112,6 +115,7 @@ describe('PipelineTasksService', () => {
           commit: 'test',
           units: [],
         });
+      expect(findOne).toBeCalled();
     });
     it('add task', async () => {
       const lpush = jest.spyOn(redisClient, 'lpush');