diff --git a/app/Jobs/Entity/CreateRawPdf.php b/app/Jobs/Entity/CreateRawPdf.php
index c5c777ae4101..df14cbba1fff 100644
--- a/app/Jobs/Entity/CreateRawPdf.php
+++ b/app/Jobs/Entity/CreateRawPdf.php
@@ -102,7 +102,6 @@ class CreateRawPdf
      */
     public function handle()
     {
-        /** Testing this override to improve PDF generation performance */
         $ps = new PdfService($this->invitation, $this->resolveType(), [
             'client' => $this->entity->client ?? false,
             'vendor' => $this->entity->vendor ?? false,
diff --git a/app/Services/Template/TemplateService.php b/app/Services/Template/TemplateService.php
index 1a890ec6e2b4..4703ffc8b77b 100644
--- a/app/Services/Template/TemplateService.php
+++ b/app/Services/Template/TemplateService.php
@@ -343,6 +343,13 @@ class TemplateService
 
     }
 
+    public function setEntity($entity): self
+    {
+        $this->entity = $entity;
+
+        return $this;
+    }
+
     /**
      * Parses all variables in the document
      *