Fixed test after fixed batch size usage.
Merged pull request #660