Fix for batch 3

Test worked for every batches except of batch 4