diff --git a/tests/App/Functional/EntityExtension/EntityExtensionTest.php b/tests/App/Functional/EntityExtension/EntityExtensionTest.php
index 24e25e1a7b..c4870b88d2 100644
--- a/tests/App/Functional/EntityExtension/EntityExtensionTest.php
+++ b/tests/App/Functional/EntityExtension/EntityExtensionTest.php
@@ -124,7 +124,7 @@ public function overwriteEntityExtensionMapInServicesInContainer(array $entityEx
 
     /**
      * @param string[] $entities
-     * @return \Doctrine\ORM\Mapping\ClassMetadata[]
+     * @return \Doctrine\Persistence\Mapping\ClassMetadata[]
      */
     public function getMetadata(array $entities): array
     {
@@ -134,7 +134,7 @@ public function getMetadata(array $entities): array
     }
 
     /**
-     * @param \Doctrine\ORM\Mapping\ClassMetadata[] $metadata
+     * @param \Doctrine\Persistence\Mapping\ClassMetadata[] $metadata
      */
     public function generateProxies(array $metadata): void
     {
@@ -142,7 +142,7 @@ public function generateProxies(array $metadata): void
     }
 
     /**
-     * @param \Doctrine\ORM\Mapping\ClassMetadata[] $metadata
+     * @param \Doctrine\Persistence\Mapping\ClassMetadata[] $metadata
      */
     public function updateDatabaseSchema(array $metadata): void
     {