diff --git a/src/Function2D.cpp b/src/Function2D.cpp
index 013bc6f..a34d1d4 100644
--- a/src/Function2D.cpp
+++ b/src/Function2D.cpp
@@ -298,7 +298,7 @@ Matrix Aurora::max(const Matrix &aMatrix, FunctionDirection direction, long& row
         default:
         {
             Eigen::Map<Eigen::MatrixXd> srcMatrix(calcMatrix.getData(),calcMatrix.getDimSize(0),calcMatrix.getDimSize(1));
-            double * ret = malloc(calcMatrix.getDimSize(0));
+            double * ret = malloc(calcMatrix.getDimSize(1));
             if (calcMatrix.getDimSize(1) == 1){
                 ret[0] = srcMatrix.col(0).maxCoeff(&rowIdx, &colIdx);
             }