Fix some UI todos.