diff --git a/src/UnitTests/Matrices/DistributedMatrixTest.h b/src/UnitTests/Matrices/DistributedMatrixTest.h
index a1a9f3eb8c281e95e4b4283d29dc69d316c01c94..9487e59776b4452f42e894ddc856100edde62b9f 100644
--- a/src/UnitTests/Matrices/DistributedMatrixTest.h
+++ b/src/UnitTests/Matrices/DistributedMatrixTest.h
@@ -171,7 +171,7 @@ TYPED_TEST( DistributedMatrixTest, getCompressedRowLengths )
 
    this->matrix.setCompressedRowLengths( this->rowLengths );
    RowLengthsVector output;
-   this->matrix.getCompressedRowLengths( output );
+   this->matrix.getCompressedRowLengths( output ); // TODO: replace this with getRowCapacities
    EXPECT_EQ( output, this->rowLengths );
 }