Bug fix
Reducing batch size