Fix UI code.