diff --git a/app/code/Magento/UrlRewrite/Model/MergeDataProvider.php b/app/code/Magento/UrlRewrite/Model/MergeDataProvider.php
index f37e6cbde7d75ebafbc092646b1dc4e383c2fa8d..20ecf0f52c05781d8acaf076cca745e04481347e 100644
--- a/app/code/Magento/UrlRewrite/Model/MergeDataProvider.php
+++ b/app/code/Magento/UrlRewrite/Model/MergeDataProvider.php
@@ -5,7 +5,7 @@
  */
 namespace Magento\UrlRewrite\Model;
 
-use Magento\UrlRewrite\Service\V1\Data\UrlRewrite;
+use Magento\UrlRewrite\Service\V1\Data\UrlRewrite as UrlRewriteService;
 
 /**
  * This class is to be used as a container for new generated url rewrites by adding new ones using merge method
@@ -24,7 +24,7 @@ class MergeDataProvider
     /**
      * Adds url rewrites to class data container by removing duplicates by a unique key
      *
-     * @param UrlRewrite[] $urlRewritesArray
+     * @param UrlRewriteService[] $urlRewritesArray
      * @return void
      */
     public function merge(array $urlRewritesArray)
@@ -42,7 +42,7 @@ class MergeDataProvider
     /**
      * Returns the data added to container
      *
-     * @return UrlRewrite[]
+     * @return UrlRewriteService[]
      */
     public function getData()
     {