diff --git a/Assignment9/src/png.c b/Assignment9/src/png.c
index 1d998d9..94af602 100644
--- a/Assignment9/src/png.c
+++ b/Assignment9/src/png.c
@@ -126,7 +126,7 @@ int get_number_of_chunks(png_chunk **chunks) {
 	while (memcmp(chunks[i]->type, "IEND", 4) != 0) {
 		i++;
 	}
-	return i;
+	return ++i;
 }
 
 void xor_data(png_chunk *chunk) {
@@ -139,8 +139,7 @@ void xor_data(png_chunk *chunk) {
 
 void destroy_chunks(png_chunk **chunks) {
 	unsigned int size = get_number_of_chunks(chunks);
-	// +1 for IEND chunk
-	for (int i = 0; i < size + 1; i++) {
+	for (int i = 0; i < size; i++) {
 		free(chunks[i]->data);
 		free(chunks[i]);
 	}