diff --git a/src/user/SharedTempStoreFactoryTrait.php b/src/user/SharedTempStoreFactoryTrait.php
index d245e20..a8745a2 100644
--- a/src/user/SharedTempStoreFactoryTrait.php
+++ b/src/user/SharedTempStoreFactoryTrait.php
@@ -12,14 +12,14 @@ trait SharedTempStoreFactoryTrait {
   /**
    * Shared tempstore factory.
    *
-   * @var \Drupal\user\SharedTempStoreFactory
+   * @var \Drupal\Core\TempStore\SharedTempStoreFactory
    */
   protected $sharedTempStoreFactory;
 
   /**
    * Sets shared tempstore factory.
    *
-   * @param \Drupal\user\SharedTempStoreFactory $file_usage
+   * @param \Drupal\Core\TempStore\SharedTempStoreFactory $file_usage
    *   Shared tempstore factory.
    *
    * @return $this
@@ -32,7 +32,7 @@ public function setSharedTempStoreFactory(SharedTempStoreFactory $file_usage) {
   /**
    * Gets shared tempstore factory.
    *
-   * @return \Drupal\user\SharedTempStoreFactory
+   * @return \Drupal\Core\TempStore\SharedTempStoreFactory
    *   Shared tempstore factory.
    */
   public function getSharedTempStoreFactory() {