Fix UI un-aligned issue