diff --git a/include/rxmesh/matrix/sparse_matrix.cuh b/include/rxmesh/matrix/sparse_matrix.cuh
index 91c94ac5..b72fc21f 100644
--- a/include/rxmesh/matrix/sparse_matrix.cuh
+++ b/include/rxmesh/matrix/sparse_matrix.cuh
@@ -537,7 +537,7 @@ struct SparseMatrix
                            rxmesh::DenseMatrix<T>& C_mat,
                            cudaStream_t            stream = 0)
     {
-        assert(cols() == B_mat.cols());
+        assert(cols() == B_mat.rows());
         assert(rows() == C_mat.rows());
         assert(B_mat.cols() == C_mat.cols());
 
diff --git a/tests/RXMesh_test/test_attribute.cuh b/tests/RXMesh_test/test_attribute.cuh
index 05deae03..2b9bfc67 100644
--- a/tests/RXMesh_test/test_attribute.cuh
+++ b/tests/RXMesh_test/test_attribute.cuh
@@ -130,10 +130,14 @@ TEST(Attribute, Reduce)
     ASSERT_EQ(cudaDeviceSynchronize(), cudaSuccess);
 
     uint32_t result = 0;
-    rx.for_each_edge(rxmesh::HOST, [&](const rxmesh::EdgeHandle eh) {
-        auto pl = eh.unpack();
-        result  = std::max(result, pl.first * pl.second);
-    });
+    rx.for_each_edge(
+        rxmesh::HOST,
+        [&](const rxmesh::EdgeHandle eh) {
+            auto pl = eh.unpack();
+            result  = std::max(result, pl.first * pl.second);
+        },
+        NULL,
+        false);
 
     EXPECT_EQ(output, result);
 }