Fix init for batch test