From 6a3319df1cbdba66dee0209f74b7c15e17a6d7b6 Mon Sep 17 00:00:00 2001
From: David Bailey <davidbailey.2889@gmail.com>
Date: Thu, 13 Apr 2023 09:59:05 +0200
Subject: [PATCH] fix: :adhesive_bandage: properly return engine from if/else

---
 lib/comfpile/core.rb | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/lib/comfpile/core.rb b/lib/comfpile/core.rb
index 73c28c8..f946511 100644
--- a/lib/comfpile/core.rb
+++ b/lib/comfpile/core.rb
@@ -61,9 +61,11 @@ module Comfpile
             new_engine = if(engine_class.is_a? Comfpile::ArtefactEngine)
                 engine_class
             else
-                engine_class.new(self,
+                engine = engine_class.new(self,
                     subpriority: @artefact_prio_counter, **options)
                 @artefact_prio_counter += 1
+
+                engine
             end
 
             yield(new_engine) if block_given?