diff --git a/Templating/UI/CountNotificationUser.php b/Templating/UI/CountNotificationUser.php
index e4d950cac..859cb673a 100644
--- a/Templating/UI/CountNotificationUser.php
+++ b/Templating/UI/CountNotificationUser.php
@@ -17,6 +17,7 @@
  */
 namespace Chill\MainBundle\Templating\UI;
 
+use Symfony\Component\Security\Core\User\UserInterface;
 use Chill\MainBundle\Entity\User;
 
 /**
@@ -37,7 +38,7 @@ class CountNotificationUser
         $this->counters[] = $counter;
     }
     
-    public function getSumNotification(User $u): int
+    public function getSumNotification(UserInterface $u): int
     {
         $sum = 0; 
         
diff --git a/Templating/UI/NotificationCounterInterface.php b/Templating/UI/NotificationCounterInterface.php
index effb7ca71..f0621b81a 100644
--- a/Templating/UI/NotificationCounterInterface.php
+++ b/Templating/UI/NotificationCounterInterface.php
@@ -17,7 +17,7 @@
  */
 namespace Chill\MainBundle\Templating\UI;
 
-use Chill\MainBundle\Entity\User;
+use Symfony\Component\Security\Core\User\UserInterface;
 
 /**
  *
@@ -28,5 +28,5 @@ interface NotificationCounterInterface
     /**
      * Add a number of notification
      */
-    public function addNotification(User $u): int;
+    public function addNotification(UserInterface $u): int;
 }