diff --git a/benchmark/BaseBench.php b/benchmark/BaseBench.php
index 8f62fdee0f..7b85fd001d 100644
--- a/benchmark/BaseBench.php
+++ b/benchmark/BaseBench.php
@@ -6,7 +6,7 @@
 
 use Doctrine\ODM\MongoDB\Configuration;
 use Doctrine\ODM\MongoDB\DocumentManager;
-use Doctrine\ODM\MongoDB\Mapping\Driver\AnnotationDriver;
+use Doctrine\ODM\MongoDB\Mapping\Driver\AttributeDriver;
 use MongoDB\Client;
 use MongoDB\Model\DatabaseInfo;
 use PhpBench\Benchmark\Metadata\Annotations\BeforeMethods;
@@ -80,8 +80,8 @@ public function clearDatabase(): void
         }
     }
 
-    protected static function createMetadataDriverImpl(): AnnotationDriver
+    protected static function createMetadataDriverImpl(): AttributeDriver
     {
-        return AnnotationDriver::create(__DIR__ . '/../tests/Documents');
+        return AttributeDriver::create(__DIR__ . '/../tests/Documents');
     }
 }