UI bug fixed

UI bug fixed