small ui fix