Issue where return to home was showing before end of battle fixed.