1 files changed, 4 insertions, 3 deletions
diff --git a/src/world.hpp b/src/world.hpp
index 7c66e6b..4ceca94 100644
--- a/src/world.hpp
+++ b/src/world.hpp
@@ -52,13 +52,14 @@ class WorldSystem : public entityx::System<WorldSystem>
 {
 private:
     std::vector<World> worlds;
-    //World& currentWorld;
+    World* currentWorld;
 public:
-    WorldSystem(void) {}
+    WorldSystem(void):
+        currentWorld(nullptr) {}
 
     ~WorldSystem(void) {}
 
-    unsigned int addWorld(sol::object);
+    World* addWorld(sol::object);
 
     /**
      * Prepares the system for running.