some UI fix