diff --git a/swh/lister/core/tasks.py b/swh/lister/core/tasks.py
index 5d4e0f6..be2ee4f 100644
--- a/swh/lister/core/tasks.py
+++ b/swh/lister/core/tasks.py
@@ -6,9 +6,8 @@ import abc
 import random
 
 from celery import group
-from celery.app.task import TaskType
 
-from swh.scheduler.task import Task
+from swh.scheduler.task import Task, TaskType
 
 from .abstractattribute import AbstractAttribute