diff --git a/src/mlpack/core/data/load_csv.hpp b/src/mlpack/core/data/load_csv.hpp
index 845d3f076f..ce7d0bb9b1 100644
--- a/src/mlpack/core/data/load_csv.hpp
+++ b/src/mlpack/core/data/load_csv.hpp
@@ -100,7 +100,7 @@ class LoadCSV
     // Reset the DatasetInfo object, if needed.
     if (info.Dimensionality() == 0)
     {
-      info = DatasetMapper<MapPolicy>(rows);
+      info.SetDimensionality(rows);
     }
     else if (info.Dimensionality() != rows)
     {
@@ -195,7 +195,7 @@ class LoadCSV
         // Reset the DatasetInfo object, if needed.
         if (info.Dimensionality() == 0)
         {
-          info = DatasetMapper<MapPolicy>(rows);
+          info.SetDimensionality(rows);
         }
         else if (info.Dimensionality() != rows)
         {