Batch-size is always 1

After !25 (merged), we now have a new bug in which the batch-size is never properly set on the data module anymore.