diff --git a/config/services/repository.yaml b/config/services/repository.yaml
index 6e8a5bd..7269f02 100644
--- a/config/services/repository.yaml
+++ b/config/services/repository.yaml
@@ -9,5 +9,4 @@ services:
         class: CleverAge\UiProcessBundle\Repository\ProcessScheduleRepository
         public: false
         arguments:
-            - '@doctrine'
-
+            - '@doctrine.orm.entity_manager'
diff --git a/src/Repository/ProcessScheduleRepository.php b/src/Repository/ProcessScheduleRepository.php
index 6c8e3ea..8aee97d 100644
--- a/src/Repository/ProcessScheduleRepository.php
+++ b/src/Repository/ProcessScheduleRepository.php
@@ -14,21 +14,21 @@
 namespace CleverAge\UiProcessBundle\Repository;
 
 use CleverAge\UiProcessBundle\Entity\ProcessSchedule;
-use Doctrine\Bundle\DoctrineBundle\Repository\ServiceEntityRepository;
-use Doctrine\Persistence\ManagerRegistry;
+use Doctrine\ORM\EntityManagerInterface;
+use Doctrine\ORM\EntityRepository;
 
 /**
- * @extends ServiceEntityRepository<ProcessSchedule>
+ * @extends EntityRepository<ProcessSchedule>
  *
  * @method ProcessSchedule|null find($id, $lockMode = null, $lockVersion = null)
  * @method ProcessSchedule|null findOneBy(mixed[] $criteria, string[] $orderBy = null)
  * @method ProcessSchedule[]    findAll()
  * @method ProcessSchedule[]    findBy(mixed[] $criteria, string[] $orderBy = null, $limit = null, $offset = null)
  */
-class ProcessScheduleRepository extends ServiceEntityRepository
+class ProcessScheduleRepository extends EntityRepository
 {
-    public function __construct(ManagerRegistry $registry)
+    public function __construct(EntityManagerInterface $em)
     {
-        parent::__construct($registry, ProcessSchedule::class);
+        parent::__construct($em, $em->getClassMetadata(ProcessSchedule::class));
     }
 }