Fixed stuck at battle end bug