diff --git a/include/types/compression.h b/include/types/compression.h
index ae1e87d..376eea9 100644
--- a/include/types/compression.h
+++ b/include/types/compression.h
@@ -62,7 +62,8 @@
 
 	int (*init)(struct comp_ctx **comp_ctx, int level);
 	int (*add_data)(struct comp_ctx *comp_ctx, const char *in_data, int in_len, struct buffer *out);
-	int (*flush)(struct comp_ctx *comp_ctx, struct buffer *out, int flag);
+	int (*flush)(struct comp_ctx *comp_ctx, struct buffer *out);
+	int (*finish)(struct comp_ctx *comp_ctx, struct buffer *out);
 	int (*reset)(struct comp_ctx *comp_ctx);
 	int (*end)(struct comp_ctx **comp_ctx);
 	struct comp_algo *next;