Fix align bug