Fixed minor battle bug