diff --git a/onetensor/testDimension.c b/onetensor/testDimension.c
index 006031e..e7cd067 100644
--- a/onetensor/testDimension.c
+++ b/onetensor/testDimension.c
@@ -64,8 +64,8 @@ TEST(SplitDim){
 
   free_dimension(D);
   free_dimension(ad);
-  free(d_part1);
-  free(d_part2);
+  free_dimension(d_part1);
+  free_dimension(d_part2);
 }
 
 TEST(SplitDim_2){
@@ -94,8 +94,8 @@ TEST(SplitDim_2){
 
   free_dimension(D);
   free_dimension(ad);
-  free(d_part1);
-  free(d_part2);
+  free_dimension(d_part1);
+  free_dimension(d_part2);
 }
 
 
@@ -259,6 +259,10 @@ TEST(list_shape_in_dim){
   dimension *dim=create_dim_from_list_shape(l_p);
 
   printDebug_dimension(dim, "from l_p"); 
+  
+  free_dimension(dim);
+  free_list_shape_in_dim(l_p);
+
 }