diff --git a/include/rxmesh/diff/diff_attribute.h b/include/rxmesh/diff/diff_attribute.h
index dfad320..7702569 100644
--- a/include/rxmesh/diff/diff_attribute.h
+++ b/include/rxmesh/diff/diff_attribute.h
@@ -1,5 +1,7 @@
 #pragma once
 
+
+
 #include "rxmesh/attribute.h"
 #include "rxmesh/diff/scalar.h"
 
@@ -21,9 +23,9 @@ class DiffAttribute : public Attribute<Scalar<T, Size, WithHessian>, HandleT>
     }
 
     explicit DiffAttribute(const char*   name,
-                           uint32_t      num_attributes,  // not used
+                           uint32_t      num_attributes,
                            locationT     location,
-                           layoutT       layout,  // not used
+                           layoutT       layout,
                            RXMeshStatic* rxmesh)
         : Attribute<ScalarType, HandleT>(name,
                                          num_attributes,
@@ -33,7 +35,6 @@ class DiffAttribute : public Attribute<Scalar<T, Size, WithHessian>, HandleT>
     {
     }
 
-
     /**
      * @brief return the value of the Scalar type as an attribute. The only
      * reason we do this is for visualization. Thus, the return attributes is
diff --git a/tests/RXMesh_test/CMakeLists.txt b/tests/RXMesh_test/CMakeLists.txt
index 33231f8..b5a5076 100644
--- a/tests/RXMesh_test/CMakeLists.txt
+++ b/tests/RXMesh_test/CMakeLists.txt
@@ -30,6 +30,7 @@ set( SOURCE_LIST
 	test_scalar.cu
 	test_diff_attribute.cu
 	test_inverse.cu
+	test_grad.h
 )
 
 target_sources( RXMesh_test 
diff --git a/tests/RXMesh_test/rxmesh_test_main.cu b/tests/RXMesh_test/rxmesh_test_main.cu
index ac53471..57c2dc7 100644
--- a/tests/RXMesh_test/rxmesh_test_main.cu
+++ b/tests/RXMesh_test/rxmesh_test_main.cu
@@ -19,6 +19,7 @@ struct RXMeshTestArg
 #include "test_patch_scheduler.cuh"
 #include "test_patch_lock.cuh"
 #include "test_wasted_work.cuh"
+#include "test_grad.h"
 // clang-format on
 
 int main(int argc, char** argv)