diff --git a/src/transmissionReconstruction/reconstruction/solvingEquationSystem/solve.cpp b/src/transmissionReconstruction/reconstruction/solvingEquationSystem/solve.cpp
index ea5dc00..fc3aaec 100644
--- a/src/transmissionReconstruction/reconstruction/solvingEquationSystem/solve.cpp
+++ b/src/transmissionReconstruction/reconstruction/solvingEquationSystem/solve.cpp
@@ -83,7 +83,7 @@ namespace Recon
                     options.TVAL3Beta0 = Recon::transParams::betaValues[i];
                     options.nonNeg = nonNeg;
                     solveResult[j] = solve(M, b, dims, transParams::maxIter, options);
-                    solveResult[j].forceReshape(dims[0], dims[1], dims.getDataSize()<3?1:dims[2]);
+                    solveResult[j].forceReshape(dims[0], dims[1], dims.getDims()<3?1:dims[2]);
                 }
                 result[i] = solveResult;
             }