Bug fixes for budget UI.