fix small ui bug