diff --git a/app/Models/Document.php b/app/Models/Document.php
index ebf0855df1b5..f2d1e43d337f 100644
--- a/app/Models/Document.php
+++ b/app/Models/Document.php
@@ -8,6 +8,14 @@ use DB;
  */
 class Document extends EntityModel
 {
+    /**
+     * @return mixed
+     */
+    public function getEntityType()
+    {
+        return ENTITY_DOCUMENT;
+    }
+
     /**
      * @var array
      */
diff --git a/app/Models/InvoiceItem.php b/app/Models/InvoiceItem.php
index 6598a24bed76..7e98ef335ed4 100644
--- a/app/Models/InvoiceItem.php
+++ b/app/Models/InvoiceItem.php
@@ -7,6 +7,14 @@ use Illuminate\Database\Eloquent\SoftDeletes;
  */
 class InvoiceItem extends EntityModel
 {
+    /**
+     * @return mixed
+     */
+    public function getEntityType()
+    {
+        return ENTITY_INVOICE_ITEM;
+    }
+
     use SoftDeletes;
     /**
      * @var array
@@ -20,7 +28,7 @@ class InvoiceItem extends EntityModel
         'tax_name1',
         'tax_rate1',
         'tax_name2',
-        'tax_rate2',    
+        'tax_rate2',
     ];
 
     /**
diff --git a/app/Models/Task.php b/app/Models/Task.php
index bc771e70e3d9..11e952420141 100644
--- a/app/Models/Task.php
+++ b/app/Models/Task.php
@@ -14,6 +14,14 @@ class Task extends EntityModel
     use SoftDeletes;
     use PresentableTrait;
 
+    /**
+     * @return mixed
+     */
+    public function getEntityType()
+    {
+        return ENTITY_TASK;
+    }
+
     /**
      * @var string
      */