little UI fix