1 files changed, 4 insertions, 7 deletions
diff --git a/sim/src/BaseBattle.h b/sim/src/BaseBattle.h
index 77e1d10..6f59457 100644
--- a/sim/src/BaseBattle.h
+++ b/sim/src/BaseBattle.h
@@ -1,11 +1,9 @@
 #pragma once
 
-#include <entt/entity/registry.hpp>
-#include <entt/signal/dispatcher.hpp>
-
 #include <kurator/engine/Context.h>
 #include <kurator/campaign/Scenario.h>
 #include <kurator/sim/Battle.h>
+#include <kurator/sim/State.h>
 
 #include "TeamManager.h"
 
@@ -20,12 +18,11 @@ class BaseBattle : public Battle
 {
 public:
 	BaseBattle(const campaign::Scenario& scenario);
-	entt::registry& registry() override;
-	entt::dispatcher& dispatcher() override;
+	engine::Context context() override;
+	engine::ConstContext const_context() const override;
 	void update(engine::Context& ctx) override;
 private:
-	entt::registry _registry;
-	entt::dispatcher _dispatcher;
+	State state;
 	TeamManager manager;
 	void pick_random_targets();
 };