Trying to fix Ui glitch

Hope so...
P.s. Sorry for the wrong branch, github did it without my knowledge!