diff --git a/Tests/DependencyInjection/JMSSerializerExtensionTest.php b/Tests/DependencyInjection/JMSSerializerExtensionTest.php
index 6b89ece5..8e5a07d8 100644
--- a/Tests/DependencyInjection/JMSSerializerExtensionTest.php
+++ b/Tests/DependencyInjection/JMSSerializerExtensionTest.php
@@ -300,11 +300,11 @@ public function testLoadExistentMetadataDir()
                 ],
             ],
         ], static function (ContainerBuilder $container) {
-            $container->getAlias('jms_serializer.metadata.file_locator')->setPublic(true);
+            $container->findDefinition('jms_serializer.metadata.file_locator')->setPublic(true);
         });
 
         $fileLocatorDef = $container->findDefinition('jms_serializer.metadata.file_locator');
-        $directories = $fileLocatorDef->getArgument(0);
+        $directories = $fileLocatorDef->getArgument(0)->getArgument(0);
         $this->assertEquals(['foo_ns' => __DIR__], $directories);
     }
 
@@ -674,7 +674,7 @@ public function testIncludeInterfaces()
         self::assertSame($expected, $actual);
     }
 
-    private function getContainerForConfigLoad(array $configs, ?callable $configurator = null)
+    private function getContainerForConfigLoad(array $configs)
     {
         $bundle = new JMSSerializerBundle();
         $extension = $bundle->getContainerExtension();