Fixed UI stuff for lolo