Small change to batch size