Fixes bug: dataset has only one dimension