fix some UI problems