Fix a UI bug