Minor ui fix