Fix output_len_lambda