Some fixes for battle mode

2 files changed, 5 insertions, 0 deletions

diff --git a/src/engine/World.cpp b/src/engine/World.cpp
index 8e20806ff..8137b3801 100644
--- a/src/engine/World.cpp
+++ b/src/engine/World.cpp
@@ -79,6 +79,10 @@ u32 World::PreviousCup() {
     return 0;
 }
 
+void World::SetCupIndex(size_t index) {
+    CupIndex = index;
+}
+
 void World::SetCup(Cup* cup) {
     if (cup) {
         CurrentCup = cup;
diff --git a/src/engine/World.h b/src/engine/World.h
index 915cacf27..bc6845aae 100644
--- a/src/engine/World.h
+++ b/src/engine/World.h
@@ -82,6 +82,7 @@ public:
 
     void AddCup(Cup*);
     void SetCup(Cup* cup);
+    void SetCupIndex(size_t index);
     const char* GetCupName();
     u32 GetCupIndex();
     u32 NextCup();