From 9977f2c47e1b86e0e70ac4db4ac4adf18d463106 Mon Sep 17 00:00:00 2001
From: David Bomba <turbo124@gmail.com>
Date: Thu, 19 Sep 2024 13:48:02 +1000
Subject: [PATCH] Minor fixes

---
 app/Jobs/Entity/CreateRawPdf.php          | 1 -
 app/Services/Template/TemplateService.php | 7 +++++++
 2 files changed, 7 insertions(+), 1 deletion(-)

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
      *