diff --git a/crypto/deflate.c b/crypto/deflate.c
index 77d7655..bc73342 100644
--- a/crypto/deflate.c
+++ b/crypto/deflate.c
@@ -93,11 +93,13 @@
 
 static void deflate_comp_exit(struct deflate_ctx *ctx)
 {
+	zlib_deflateEnd(&ctx->comp_stream);
 	vfree(ctx->comp_stream.workspace);
 }
 
 static void deflate_decomp_exit(struct deflate_ctx *ctx)
 {
+	zlib_inflateEnd(&ctx->decomp_stream);
 	kfree(ctx->decomp_stream.workspace);
 }