small UI fix