UI change
little bug fix