diff --git a/src/Controller/ImageStyleDownloadController.php b/src/Controller/ImageStyleDownloadController.php
index df7b195..c0b96e1 100644
--- a/src/Controller/ImageStyleDownloadController.php
+++ b/src/Controller/ImageStyleDownloadController.php
@@ -83,11 +83,11 @@ class ImageStyleDownloadController extends CoreImageStyleDownloadController {
   /**
    * Builds the response.
    */
-  public function deliver(Request $request, $scheme, ImageStyleInterface $image_style) {
+  public function deliver(Request $request, $scheme, ImageStyleInterface $image_style, string $required_derivative_scheme) {
     // Interesting just the defined in JPEG_PATTERN extensions.
     if (!preg_match(ImageFactoryInterface::JPG_PATTERN, $request->getPathInfo())) {
       // Don't match with extensions go to system image style delivery method.
-      return parent::deliver($request, $scheme, $image_style);
+      return parent::deliver($request, $scheme, $image_style, $required_derivative_scheme);
     }
 
     // Prepare image uri.
@@ -99,7 +99,7 @@ class ImageStyleDownloadController extends CoreImageStyleDownloadController {
     $derivative_uri = $image_style->buildUri($image_uri);
     if (!preg_match('/.webp$/', $derivative_uri)) {
       // The image style is not converted to webp format go to system deliver.
-      return parent::deliver($request, $scheme, $image_style);
+      return parent::deliver($request, $scheme, $image_style, $required_derivative_scheme);
     }
 
     $headers = [];
diff --git a/src/Routing/RouteSubscriber.php b/src/Routing/RouteSubscriber.php
index e478c97..b450df5 100644
--- a/src/Routing/RouteSubscriber.php
+++ b/src/Routing/RouteSubscriber.php
@@ -16,7 +16,10 @@ class RouteSubscriber extends RouteSubscriberBase {
   protected function alterRoutes(RouteCollection $collection): void {
     foreach (['image.style_public', 'image.style_private'] as $route_id) {
       if ($route = $collection->get($route_id)) {
-        $route->setDefault('_controller', 'Drupal\wpf\Controller\ImageStyleDownloadController::deliver');
+        $route->setDefault(
+          '_controller', 'Drupal\wpf\Controller\ImageStyleDownloadController::deliver',
+          'required_derivative_scheme', 'public'
+        );
       }
     }
   }