diff --git a/src/AutoIncrementID.php b/src/AutoIncrementID.php
index d675c9d..1312238 100644
--- a/src/AutoIncrementID.php
+++ b/src/AutoIncrementID.php
@@ -2,6 +2,10 @@
 
 namespace ETNA\Doctrine\Extensions;
 
+use Doctrine\ORM\Mapping\Id;
+use Doctrine\ORM\Mapping\GeneratedValue;
+use Doctrine\ORM\Mapping\Column;
+
 trait AutoIncrementID
 {
     /**
diff --git a/src/CreatedAt.php b/src/CreatedAt.php
index ca8898b..5589471 100644
--- a/src/CreatedAt.php
+++ b/src/CreatedAt.php
@@ -2,6 +2,9 @@
 
 namespace ETNA\Doctrine\Extensions;
 
+use Doctrine\ORM\Mapping\Column;
+use Doctrine\ORM\Mapping\PrePersist;
+
 trait CreatedAt
 {
     /**
diff --git a/src/DoNotDelete.php b/src/DoNotDelete.php
index 2aeaa68..76fa887 100644
--- a/src/DoNotDelete.php
+++ b/src/DoNotDelete.php
@@ -2,6 +2,9 @@
 
 namespace ETNA\Doctrine\Extensions;
 
+use Doctrine\ORM\Mapping\Column;
+use Doctrine\ORM\Mapping\PreRemove;
+
 trait DoNotDelete
 {
     /**
@@ -43,7 +46,7 @@ public function setDeletedAt($datetime)
     }
 
     /**
-     * @preRemove
+     * @PreRemove
      */
     public function canNotDelete()
     {
diff --git a/src/UpdatedAt.php b/src/UpdatedAt.php
index 6099e5c..92adb71 100644
--- a/src/UpdatedAt.php
+++ b/src/UpdatedAt.php
@@ -2,6 +2,10 @@
 
 namespace ETNA\Doctrine\Extensions;
 
+use Doctrine\ORM\Mapping\Column;
+use Doctrine\ORM\Mapping\PrePersist;
+use Doctrine\ORM\Mapping\PreUpdate;
+
 trait UpdatedAt
 {
     /**