Tiny UI fix