Add max_align hack