Fix math UI bugs