Fix UI issue in header