diff --git a/src/Template/Bake/Model/entity.ctp b/src/Template/Bake/Model/entity.ctp
index 6fe25f6ce..ad4391423 100644
--- a/src/Template/Bake/Model/entity.ctp
+++ b/src/Template/Bake/Model/entity.ctp
@@ -13,8 +13,6 @@
  * @license       http://www.opensource.org/licenses/mit-license.php MIT License
  */
 
-use Cake\ORM\Association;
-
 $propertyHintMap = null;
 if (!empty($propertySchema)) {
     $propertyHintMap = $this->DocBlock->buildEntityPropertyHintTypeMap($propertySchema);
diff --git a/tests/comparisons/Model/testBakeEntityWithPropertyTypeHints.php b/tests/comparisons/Model/testBakeEntityWithPropertyTypeHints.php
index 0cdd20177..cc04d6a68 100644
--- a/tests/comparisons/Model/testBakeEntityWithPropertyTypeHints.php
+++ b/tests/comparisons/Model/testBakeEntityWithPropertyTypeHints.php
@@ -12,8 +12,8 @@
  * @property string $title
  * @property string $body
  * @property bool $published
- * @property string|int|\DateTime|\Cake\I18n\Time $created
- * @property string|int|\DateTime|\Cake\I18n\Time $updated
+ * @property \Cake\I18n\Time $created
+ * @property \Cake\I18n\Time $updated
  * @property $unknown_type
  * @property \BakeTest\Model\Entity\Author[] $authors
  */