Fixing some UI things.