Slight UI adjustment