Fix bug in loading multi-dimensional data.