Another minor ui fix