diff --git a/app/Models/BaseModel.php b/app/Models/BaseModel.php
index d6ea195907cc..a1cb856d2cb0 100644
--- a/app/Models/BaseModel.php
+++ b/app/Models/BaseModel.php
@@ -13,7 +13,6 @@ namespace App\Models;
 
 use App\DataMapper\ClientSettings;
 use App\DataMapper\CompanySettings;
-use App\Designs\Designer;
 use App\Filters\QueryFilters;
 use App\Models\Design;
 use App\Utils\Traits\MakesHash;
@@ -176,15 +175,6 @@ class BaseModel extends Model
             ->where('id', $this->decodePrimaryKey($value))->firstOrFail();
     }
 
-    public function getEntityDesigner()
-    {
-        $design = Design::find($this->decodePrimaryKey($this->client->getSetting('invoice_design_id')));
-
-        $entity = strtolower(class_basename($this));
-
-        return new Designer($this, $design, $this->client->getSetting('pdf_variables'), $entity);
-    }
-
     /**
      * @return string
      */
diff --git a/app/Repositories/ActivityRepository.php b/app/Repositories/ActivityRepository.php
index 349e47db1962..ec1de4df14e8 100644
--- a/app/Repositories/ActivityRepository.php
+++ b/app/Repositories/ActivityRepository.php
@@ -115,7 +115,7 @@ class ActivityRepository extends BaseRepository
         $entity_design_id = $entity->design_id ? $entity->design_id : $this->decodePrimaryKey($entity->client->getSetting($entity_design_id));
 
         $design = Design::find($entity_design_id);
-        $html = new HtmlEngine($invitation);
+        $html = new HtmlEngine($entity->invitations->first());
 
         if ($design->is_custom) {
           $options = [
@@ -147,5 +147,5 @@ class ActivityRepository extends BaseRepository
                      ->getCompiledHTML(true);
 
     }
-    
+
 }