Fix UI display bug for warnings