Fix leaving battles after they end