diff --git a/tests/unit/Core/Media/MediaUrlGeneratorTest.php b/tests/unit/Core/Media/MediaUrlGeneratorTest.php
index d69abf6..ea650b1 100644
--- a/tests/unit/Core/Media/MediaUrlGeneratorTest.php
+++ b/tests/unit/Core/Media/MediaUrlGeneratorTest.php
@@ -37,9 +37,9 @@ public function testGenerateWithUrlParam(): void
             'ExtensionsAllowList' => 'jpg',
             'ThumbnailPattern' => '{mediaUrl}/{mediaPath}?width={width}',
         ];
-        $configReader->expects($this->any())
+        $configReader
             ->method('getConfig')
-            ->willReturnCallback(function ($key) use ($defaultPluginConfig) {
+            ->willReturnCallback(function (string $key) use ($defaultPluginConfig) {
                 return $defaultPluginConfig[$key] ?? null;
             });
 
@@ -84,9 +84,9 @@ public function testGenerateWithExtendedUrlParam(): void
             'ExtensionsAllowList' => 'jpg',
             'ThumbnailPattern' => '{mediaUrl}/{mediaPath}?width={width}',
         ];
-        $configReader->expects($this->any())
+        $configReader
             ->method('getConfig')
-            ->willReturnCallback(function ($key) use ($defaultPluginConfig) {
+            ->willReturnCallback(function (string $key) use ($defaultPluginConfig) {
                 return $defaultPluginConfig[$key] ?? null;
             });
 
@@ -141,9 +141,9 @@ public function testGenerateWithInactiveConfig(): void
             'ExtensionsAllowList' => 'jpg',
             'ThumbnailPattern' => '{mediaUrl}/{mediaPath}?width={width}',
         ];
-        $configReader->expects($this->any())
+        $configReader
             ->method('getConfig')
-            ->willReturnCallback(function ($key) use ($defaultPluginConfig) {
+            ->willReturnCallback(function (string $key) use ($defaultPluginConfig) {
                 return $defaultPluginConfig[$key] ?? null;
             });
 
@@ -195,9 +195,9 @@ public function testGenerateWithNotAllowedExtensionResultsInOriginal(string $all
             'ExtensionsAllowList' => $allowList,
             'ThumbnailPattern' => '{mediaUrl}/{mediaPath}?width={width}',
         ];
-        $configReader->expects($this->any())
+        $configReader
             ->method('getConfig')
-            ->willReturnCallback(function ($key) use ($defaultPluginConfig) {
+            ->willReturnCallback(function (string $key) use ($defaultPluginConfig) {
                 return $defaultPluginConfig[$key] ?? null;
             });
 
@@ -244,7 +244,7 @@ public function testMultipleGenerationsUseCachedExtensionAllowList(): void
         ];
         $configReader->expects($this->exactly(3))
             ->method('getConfig')
-            ->willReturnCallback(function ($key) use ($defaultPluginConfig) {
+            ->willReturnCallback(function (string $key) use ($defaultPluginConfig) {
                 return $defaultPluginConfig[$key] ?? null;
             });