diff --git a/app/code/Magento/CatalogUrlRewrite/Model/CategoryUrlRewriteGenerator.php b/app/code/Magento/CatalogUrlRewrite/Model/CategoryUrlRewriteGenerator.php
index e565760f33448346c7434b1bb4b2fc816dbd173d..84ecb5b8a53e89880d772abefaec10356f727cfa 100644
--- a/app/code/Magento/CatalogUrlRewrite/Model/CategoryUrlRewriteGenerator.php
+++ b/app/code/Magento/CatalogUrlRewrite/Model/CategoryUrlRewriteGenerator.php
@@ -79,7 +79,7 @@ class CategoryUrlRewriteGenerator
      * @param int|null $rootCategoryId
      * @return \Magento\UrlRewrite\Service\V1\Data\UrlRewrite[]
      */
-    public function generate(Category $category, $overrideStoreUrls = false, $rootCategoryId = null)
+    public function generate($category, $overrideStoreUrls = false, $rootCategoryId = null)
     {
         if ($rootCategoryId === null) {
             $rootCategoryId = $category->getEntityId();