fix end battle