diff --git a/src/UnicaenAuthentification/Service/Traits/UserContextServiceAwareTrait.php b/src/UnicaenAuthentification/Service/Traits/UserContextServiceAwareTrait.php
index 6a56a0b606b8d6a3fb1d051c5b459a680013390f..dda6f4e2557383b9c95dd50d4ee4f7999ef3cc0d 100644
--- a/src/UnicaenAuthentification/Service/Traits/UserContextServiceAwareTrait.php
+++ b/src/UnicaenAuthentification/Service/Traits/UserContextServiceAwareTrait.php
@@ -18,10 +18,15 @@ trait UserContextServiceAwareTrait
* @param UserContext $serviceUserContext
* @return self
*/
- public function setServiceUserContext(UserContext $serviceUserContext)
+ public function setServiceUserContext(UserContext $serviceUserContext) : UserContext
{
$this->serviceUserContext = $serviceUserContext;
+ return $this->serviceUserContext;
+ }
- return $this;
+ public function getServiceUserContext() : UserContext
+ {
+ return $this->serviceUserContext;
}
-}
\ No newline at end of file
+
+}