fixed some ui shit