Fix UI breakage.