tiny UI fix