Fix bug with budget calculation