Remove update_budget method from flight.