Fixed ugly UI error