diff --git a/src/Infrastructure/Application/Service/DoctrineSession.php b/src/Infrastructure/Application/Service/DoctrineSession.php
index 0d934bf..76d5e52 100755
--- a/src/Infrastructure/Application/Service/DoctrineSession.php
+++ b/src/Infrastructure/Application/Service/DoctrineSession.php
@@ -4,6 +4,7 @@
 
 use Ddd\Application\Service\TransactionalSession;
 use Doctrine\ORM\EntityManager;
+use Doctrine\ORM\EntityManagerInterface;
 
 /**
  * Class DoctrineSession
@@ -17,9 +18,9 @@ class DoctrineSession implements TransactionalSession
     private $entityManager;
 
     /**
-     * @param EntityManager $entityManager
+     * @param EntityManagerInterface $entityManager
      */
-    public function __construct(EntityManager $entityManager)
+    public function __construct(EntityManagerInterface $entityManager)
     {
         $this->entityManager = $entityManager;
     }