Fixed UI bug yay