Fix bug with I/O ensemble size