1 files changed, 5 insertions, 5 deletions
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;
     };
 }