Minor budget refactor