Minor change on UI