diff --git a/src/logistic.c b/src/logistic.c
index b664326c..49538635 100644
--- a/src/logistic.c
+++ b/src/logistic.c
@@ -27,10 +27,8 @@ void softmax_matrix(double_matrix_t *matrix) {
     size_t num_rows = matrix->m;
     size_t num_cols = matrix->n;
 
-    double *values = matrix->values;
-
     for (int i = 0; i < num_rows; i++) {
+        double *values = double_matrix_get_row(matrix, i);
         softmax_vector(values, num_cols);
-        values += num_cols;
     }
 }
\ No newline at end of file