Fix batch test