bug fix for ui