Fix UI

Fix some UI