Fixed a little UI stuff