Forgot to set battle for player after starting cocoon

src/main/java/emu/lunarcore/game/battle/BattleService.java

@@ -148,6 +148,7 @@ public class BattleService extends BaseGameService {
         
         // Build battle from cocoon data
         Battle battle = new Battle(player, player.getLineupManager().getCurrentLineup(), stages);
+        player.setBattle(battle);
         
         // Send packet
         player.sendPacket(new PacketStartCocoonStageScRsp(battle, cocoonId, wave));