Fix ALIGN_WITH_INPUT