From 4de7887c3541459c1e9bb1c222e22c9fcc1d1c29 Mon Sep 17 00:00:00 2001
From: dutchie031 <timrorije@gmail.com>
Date: Wed, 7 May 2025 14:47:47 +0200
Subject: [PATCH] bluesam reference

---
 classes/stageClasses/Stages/PrimaryStage.lua | 5 +++--
 classes/stageClasses/Stages/WaitingStage.lua | 5 +++--
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/classes/stageClasses/Stages/PrimaryStage.lua b/classes/stageClasses/Stages/PrimaryStage.lua
index 568d084..c8681e7 100644
--- a/classes/stageClasses/Stages/PrimaryStage.lua
+++ b/classes/stageClasses/Stages/PrimaryStage.lua
@@ -3,8 +3,6 @@
 local PrimaryStage = {}
 
 PrimaryStage.__index = PrimaryStage
-local Stage = Spearhead.classes.stageClasses.Stages.BaseStage.Stage
-setmetatable(PrimaryStage, Stage)
 
 ---comment
 ---@param database Database
@@ -14,6 +12,9 @@ setmetatable(PrimaryStage, Stage)
 ---@return PrimaryStage
 function PrimaryStage.New(database, stageConfig, logger, initData)
 
+    local Stage = Spearhead.classes.stageClasses.Stages.BaseStage.Stage
+    setmetatable(PrimaryStage, Stage)
+
     local self = setmetatable({}, { __index = PrimaryStage }) --[[@as PrimaryStage]]
     self:superNew(database, stageConfig, logger, initData, "primary")
     return self
diff --git a/classes/stageClasses/Stages/WaitingStage.lua b/classes/stageClasses/Stages/WaitingStage.lua
index 163f773..d5a5bde 100644
--- a/classes/stageClasses/Stages/WaitingStage.lua
+++ b/classes/stageClasses/Stages/WaitingStage.lua
@@ -5,8 +5,6 @@
 local WaitingStage = {}
 
 WaitingStage.__index = WaitingStage
-local Stage = Spearhead.classes.stageClasses.Stages.BaseStage.Stage
-setmetatable(WaitingStage, Stage)
 
 ---@class WaitingStageInitData : StageInitData
 ---@field waitingSeconds integer
@@ -20,6 +18,9 @@ local WaitingStageInitData = {}
 ---@return WaitingStage
 function WaitingStage.New(database, stageConfig, logger, initData)
 
+    local Stage = Spearhead.classes.stageClasses.Stages.BaseStage.Stage
+    setmetatable(WaitingStage, Stage)
+
     local self = setmetatable({}, { __index = WaitingStage }) --[[@as WaitingStage]]
     self:superNew(database, stageConfig, logger, initData, "none")