Fixed minor UI bug