Fix UI (broke tests).