Fix error: ValueError: The number of fields in the new order differs from the number of fields in the schema.