diff --git a/Reflection/EntityMetadataInspector.php b/Reflection/EntityMetadataInspector.php
index 4fb877988d..efa98cc9b6 100644
--- a/Reflection/EntityMetadataInspector.php
+++ b/Reflection/EntityMetadataInspector.php
@@ -11,6 +11,7 @@
 
 namespace JavierEguiluz\Bundle\EasyAdminBundle\Reflection;
 
+use Doctrine\Common\Persistence\Mapping\ClassMetadata;
 use Symfony\Bridge\Doctrine\ManagerRegistry;
 
 /**
@@ -31,7 +32,7 @@ public function __construct(ManagerRegistry $manager)
      *
      * @param string $entityClass
      *
-     * @return array
+     * @return ClassMetadata
      */
     public function getEntityMetadata($entityClass)
     {