diff --git a/sk1/compressor.c b/sk1/compressor.c
index 10db640..e04927a 100644
--- a/sk1/compressor.c
+++ b/sk1/compressor.c
@@ -217,7 +217,7 @@ int compressFile(const char* input_file_name, const char* output_file_name) {
         byte <<= (8 - bitCount);
         fwrite(&byte, sizeof(unsigned char), 1, outputFile);
     }
-
+    freeHuffmanTree(root);
     fclose(inputFile);
     fclose(outputFile);
     return 0;
@@ -270,7 +270,7 @@ int decompressFile(const char* input_file_name, const char* output_file_name) {
             }
         }
     }
-
+    freeHuffmanTree(root);
     fclose(inputFile);
     fclose(outputFile);
     return 0;