1 files changed, 5 insertions, 5 deletions
diff --git a/src/core/state_manager.h b/src/core/state_manager.h
index 7d5e734..469c48f 100644
--- a/src/core/state_manager.h
+++ b/src/core/state_manager.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 on_event(Event& event);
         
-        void register_state(const std::string& state_name, std::unique_ptr<State> state);
+        void register_state(const std::string& state_name, Scope<State> state);
         void switch_to_state(const std::string& state_name);
 
         std::string get_current_state_name() const { return m_current_state_name; }
@@ -34,6 +34,6 @@ namespace Donut
         State*      m_current_state     = nullptr;
         std::string m_current_state_name = "";
         
-        std::unordered_map<std::string, std::unique_ptr<State>> m_states;
+        std::unordered_map<std::string, Scope<State>> m_states;
     };
 }