Fixed minor UI error