Fix align code