1 files changed, 5 insertions, 4 deletions
diff --git a/lib/algo/avl_tree.c b/lib/algo/avl_tree.c
index cc06254..d8f887c 100644
--- a/lib/algo/avl_tree.c
+++ b/lib/algo/avl_tree.c
@@ -1,4 +1,5 @@
 #include <lib/algo/avl_tree.h>
+#include <lib/mem/galloc.h>
 #include <lib/monad.h>
 
 #include <stddef.h>
@@ -55,7 +56,7 @@ AVLNode *left_rotate(AVLNode *parent) {
 
 // Create AVL node
 AVLNode *create_avl_node(void *data, AvlComparator compare) {
-  AVLNode *node = (AVLNode *)malloc(sizeof(AVLNode));
+  AVLNode *node = (AVLNode *)g_malloc(sizeof(AVLNode));
   if (node == NULL) {
     return NULL;
   }
@@ -134,7 +135,7 @@ void free_avl_tree(AVLNode *root) {
   if (root != NULL) {
     free_avl_tree(root->left);
     free_avl_tree(root->right);
-    free(root);
+    g_free(root);
   }
 }
 
@@ -144,8 +145,8 @@ void free_avl_tree_nodes(AVLNode *root) {
     free_avl_tree_nodes(root->left);
     free_avl_tree_nodes(root->right);
     if (root->data != NULL) {
-      free(root->data);
+      g_free(root->data);
     }
-    free(root);
+    g_free(root);
   }
 }