Fixed test after fixed batch size usage.