diff --git a/src/AssociationHydrator.php b/src/AssociationHydrator.php
index 9794dda..de4b744 100644
--- a/src/AssociationHydrator.php
+++ b/src/AssociationHydrator.php
@@ -5,8 +5,8 @@
 namespace SyliusLabs\AssociationHydrator;
 
 use Doctrine\Common\Collections\Collection;
-use Doctrine\Common\Persistence\Mapping\ClassMetadata;
 use Doctrine\ORM\EntityManager;
+use Doctrine\Persistence\Mapping\ClassMetadata;
 use Symfony\Component\PropertyAccess\PropertyAccess;
 use Symfony\Component\PropertyAccess\PropertyAccessor;