UI issue after new update