diff --git a/src/Entity/TaskEntity.php b/src/Entity/TaskEntity.php
index af86926..3b6b074 100644
--- a/src/Entity/TaskEntity.php
+++ b/src/Entity/TaskEntity.php
@@ -1,6 +1,7 @@
 <?php
 namespace App\Entity;
 
+use App\Exception\InvalidEnumKeyException;
 use App\Enum\StatusEnum;
 use App\Helper\StringGenerationHelper;
 use DateTime;
@@ -196,7 +197,7 @@
     }
     
     /**
-     * Get the status of the activity
+     * Get the status of the task
      *
      * @return string
      * @see StatusEnum
@@ -207,7 +208,7 @@
     }
     
     /**
-     * Set a new status for the activity
+     * Set a new status for the task
      *
      * @param string $status
      * @return self