diff --git a/src/initialize-transformed.js b/src/initialize-transformed.js
index 6ec9a3c..bd276d8 100644
--- a/src/initialize-transformed.js
+++ b/src/initialize-transformed.js
@@ -163,7 +163,7 @@ function generateDataSet (
 
   // Make sure all rows are saturated, otherwise data risks being displayed in the wrong column
   matrix = matrix.map((row, rowIndex) => {
-    if ((hasSecondDimension && row.length == dimension2.length)
+    if ((hasSecondDimension && row.length == (dimension2.length * measurements.length))
       || (!hasSecondDimension && row.length == measurements.length)) {
       // Row is saturated
       return row;