MInor UI fix!