Fixed UI error