1 files changed, 4 insertions, 3 deletions
diff --git a/src/position_stack.c b/src/position_stack.c
index 846d933..597ed5a 100644
--- a/src/position_stack.c
+++ b/src/position_stack.c
@@ -1,4 +1,5 @@
 #include "position_stack.h"
+#include "utils.h"
 #include <stdio.h>
 #include <stdlib.h>
 
@@ -19,13 +20,13 @@ void stack_error(int err)
 }
 
 // Creates a new stack
-PositionStack *pos_stack_create(int capacity)
+PositionStack *pos_stack_create(unsigned int capacity)
 {
-	PositionStack *pos_stack = malloc(sizeof(PositionStack));
+	PositionStack *pos_stack = malloc_s(sizeof(PositionStack));
 
 	pos_stack->capacity = capacity;
 	pos_stack->top = -1;
-	pos_stack->items = malloc(sizeof(Position) * capacity);
+	pos_stack->items = malloc_s(sizeof(Position) * capacity);
 
 	return pos_stack;
 }