diff --git a/src/Service/RegenerateProductUrl.php b/src/Service/RegenerateProductUrl.php
index 9fe4dac..d9b53f0 100644
--- a/src/Service/RegenerateProductUrl.php
+++ b/src/Service/RegenerateProductUrl.php
@@ -46,7 +46,7 @@ class RegenerateProductUrl
      */
     public function __construct(
         CollectionFactory $collectionFactory,
-        ProductUrlRewriteGenerator $urlRewriteGenerator,
+        ProductUrlRewriteGenerator\Proxy $urlRewriteGenerator,
         UrlPersistInterface $urlPersist,
         StoreManagerInterface $storeManager
     ) {