diff --git a/app/Models/Contact.php b/app/Models/Contact.php
index fbac37117e13..feece59de005 100644
--- a/app/Models/Contact.php
+++ b/app/Models/Contact.php
@@ -18,6 +18,14 @@ class Contact extends EntityModel implements AuthenticatableContract, CanResetPa
      */
     protected $dates = ['deleted_at'];
 
+    /**
+     * @return mixed
+     */
+    public function getEntityType()
+    {
+        return ENTITY_CONTACT;
+    }
+
     /**
      * @var array
      */
diff --git a/app/Models/Invitation.php b/app/Models/Invitation.php
index d2d4abb026fd..a86cc721735f 100644
--- a/app/Models/Invitation.php
+++ b/app/Models/Invitation.php
@@ -15,6 +15,14 @@ class Invitation extends EntityModel
      */
     protected $dates = ['deleted_at'];
 
+    /**
+     * @return mixed
+     */
+    public function getEntityType()
+    {
+        return ENTITY_INVITATION;
+    }
+
     /**
      * @return mixed
      */