Update lengths to changes made in new_masking branch.