Fixed slight ui problem