From e050dce79b67ae6386240567a994454711ab998d Mon Sep 17 00:00:00 2001
From: hachem <im@hachem.wtf>
Date: Thu, 14 Aug 2025 00:42:05 +0200
Subject: [add]: Custom memory managers and fix includes

---
 src/Core/StateManager.h | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

(limited to 'src/Core/StateManager.h')

diff --git a/src/Core/StateManager.h b/src/Core/StateManager.h
index 06370f3..c0b2b13 100644
--- a/src/Core/StateManager.h
+++ b/src/Core/StateManager.h
@@ -1,11 +1,11 @@
 #pragma once
 
-#include <memory>
+#include "Memory.h"
+#include "State.h"
+
 #include <string>
 #include <unordered_map>
 
-#include "State.h"
-
 namespace Donut
 {
     class StateManager
@@ -18,7 +18,7 @@ namespace Donut
         void Render();
         void OnEvent(Event& event);
         
-        void RegisterState(const std::string& stateName, std::unique_ptr<State> state);
+        void RegisterState(const std::string& stateName, Scope<State> state);
         void SwitchToState(const std::string& stateName);
 
         std::string GetCurrentStateName() const { return m_CurrentStateName; }
@@ -34,6 +34,6 @@ namespace Donut
         State*      m_CurrentState     = nullptr;
         std::string m_CurrentStateName = "";
         
-        std::unordered_map<std::string, std::unique_ptr<State>> m_States;
+        std::unordered_map<std::string, Scope<State>> m_States;
     };
 }
-- 
cgit v1.3