passing batch_index + 1 which is actually the correct batch number