A fix for Batch extension

Signed-off-by: Antanas Norkus <antanas.norkus@cern.ch>