diff --git a/src/EntityManagerBuilder.php b/src/EntityManagerBuilder.php
index 6d2fbed..cc751fe 100644
--- a/src/EntityManagerBuilder.php
+++ b/src/EntityManagerBuilder.php
@@ -17,7 +17,7 @@
 use Doctrine\ORM\Configuration;
 use Doctrine\ORM\EntityManager;
 use Doctrine\ORM\Mapping\DefaultQuoteStrategy;
-use Doctrine\ORM\Mapping\Driver\StaticPHPDriver;
+use Doctrine\Common\Persistence\Mapping\Driver\StaticPHPDriver;
 use Doctrine\ORM\Mapping\NamingStrategy;
 use Doctrine\ORM\Mapping\QuoteStrategy;
 use Doctrine\ORM\Mapping\UnderscoreNamingStrategy;