fixed UI problem