Minor UI fix