Fix align bugs