small fix for ui