diff --git a/phpunit.xml.dist b/phpunit.xml.dist
index 003dc7b2..de308577 100644
--- a/phpunit.xml.dist
+++ b/phpunit.xml.dist
@@ -11,15 +11,15 @@
     beStrictAboutCoverageMetadata="true"
     displayDetailsOnTestsThatTriggerDeprecations="true"
 >
-    <coverage>
-        <include>
-            <directory suffix=".php">src</directory>
-        </include>
-    </coverage>
-
     <testsuites>
         <testsuite name="DI Builder Test Suite">
             <directory>test</directory>
         </testsuite>
     </testsuites>
+
+    <source>
+        <include>
+            <directory suffix=".php">src</directory>
+        </include>
+    </source>
 </phpunit>
diff --git a/test/ContainerBuilderTest.php b/test/ContainerBuilderTest.php
index 18cc9d1f..5e707f18 100644
--- a/test/ContainerBuilderTest.php
+++ b/test/ContainerBuilderTest.php
@@ -26,15 +26,14 @@
 #[PHPUnit\UsesClass(Xml::class)]
 final class ContainerBuilderTest extends TestCase
 {
-    /** @var Generator&MockObject */
-    private Generator $generator;
+    private Generator&MockObject $generator;
     private ContainerConfiguration $config;
     private ParameterBag $parameterBag;
 
     #[PHPUnit\Before]
     public function configureDependencies(): void
     {
-        $this->generator    = $this->getMockForAbstractClass(Generator::class, [], '', false, true, true, ['generate']);
+        $this->generator    = $this->createMock(Generator::class);
         $this->config       = new ContainerConfiguration('Me\\MyApp');
         $this->parameterBag = new ParameterBag();
     }
@@ -105,7 +104,7 @@ public function constructShouldReceiveTheDependenciesAsArguments(): void
     public function setGeneratorShouldChangeTheAttributeAndReturnSelf(): void
     {
         $builder   = new ContainerBuilder($this->config, $this->generator, $this->parameterBag);
-        $generator = $this->getMockForAbstractClass(Generator::class, [], '', false);
+        $generator = $this->createMock(Generator::class);
         $expected  = new ContainerBuilder($this->config, $generator, $this->parameterBag);
 
         // @phpstan-ignore-next-line method is deprecated and will be removed in the next major version
diff --git a/test/GeneratorTest.php b/test/GeneratorTest.php
index 4f3af354..1ed474c0 100644
--- a/test/GeneratorTest.php
+++ b/test/GeneratorTest.php
@@ -40,10 +40,7 @@ public function initializeContainerShouldAddTheConfigurationFileAsAResource(): v
     #[PHPUnit\Test]
     public function initializeContainerCanOptionallyUseACustomClass(): void
     {
-        $generator = $this->getMockForAbstractClass(
-            Generator::class,
-            [__FILE__, CustomContainerBuilderForTests::class],
-        );
+        $generator = new Yaml(__FILE__, CustomContainerBuilderForTests::class);
 
         self::assertInstanceOf(
             CustomContainerBuilderForTests::class,