fix some UI stuff