fix make_partition.sh for new schema