diff --git a/src/buffer.cpp b/src/buffer.cpp
index 864f404..d502600 100644
--- a/src/buffer.cpp
+++ b/src/buffer.cpp
@@ -1,11 +1,12 @@
 #include "buffer.h"
 
-void buffer::write(char *data_in, int data_size)
+void buffer::write(char *data_in, size_t data_size)
 {
 	if (data_size > allocated - size)
 	{
 		allocated += (size + data_size) + BUFFER_SIZE;
 		data = (char *)realloc(data, allocated);
+		allocations++;
 		if (!data)
 		{
 			throw std::runtime_error("Allocation failed");
diff --git a/src/buffer.h b/src/buffer.h
index 9ed33ec..4ecb324 100644
--- a/src/buffer.h
+++ b/src/buffer.h
@@ -9,8 +9,9 @@ struct buffer
 	char *data;
 	size_t size;
 	size_t allocated;
+	int allocations;
 
-	void write(char *data_in, int data_size);
+	void write(char *data_in, size_t data_size);
 	void remove(int offset, int remove_size);
 
 	~buffer()