make_batch.py fixed