Fix ugly UI with super bad hack