Fixed bug in the budget