There was a bug when generating all_all mlp