1 files changed, 3 insertions, 2 deletions
diff --git a/lib/seg/mask_data.c b/lib/seg/mask_data.c
index 144c7ca..5789f08 100644
--- a/lib/seg/mask_data.c
+++ b/lib/seg/mask_data.c
@@ -1,3 +1,4 @@
+#include <lib/mem/galloc.h>
 #include <lib/seg/mask_data.h>
 #include <lib/seg/util.h>
 
@@ -5,7 +6,7 @@
 
 // Allocate Mask Data for Label
 MaskData *create_mask_data(MaskData_t label) {
-  MaskData *data = (MaskData *)malloc(sizeof(MaskData));
+  MaskData *data = (MaskData *)g_malloc(sizeof(MaskData));
   data->label = label;
   data->area = 0;
   data->perimeter = 0;
@@ -33,7 +34,7 @@ AVLNode *insert_mask_alloc(AVLNode *node, MaskData_t label) {
   MaskData *data = create_mask_data(label);
   Result result = insert_mask(node, data);
   if (!result.success) {
-    free(data);
+    g_free(data);
   }
   return (AVLNode *)result.data;
 }