Fix a minor UI broken issue