Fixing some UI stuff.