bug fix for UI