Fix the batchsize