fix a UI bug caused by the last refactory