Slight change to max_lens in Amortize