Correct checking for only_alignment flag