diff --git a/include/libssh/crypto.h b/include/libssh/crypto.h
index a5111802..dad5883d 100644
--- a/include/libssh/crypto.h
+++ b/include/libssh/crypto.h
@@ -231,6 +231,8 @@ int secure_memcmp(const void *s1, const void *s2, size_t n);
 ENGINE *pki_get_engine(void);
 #endif /* HAVE_LIBCRYPTO */
 
+void compress_cleanup(struct ssh_crypto_struct *crypto);
+
 #ifdef __cplusplus
 }
 #endif
diff --git a/src/gzip.c b/src/gzip.c
index e2d2ac2f..27c8042b 100644
--- a/src/gzip.c
+++ b/src/gzip.c
@@ -260,3 +260,17 @@ decompress_buffer(ssh_session session, ssh_buffer buf, size_t maxlen)
     SSH_BUFFER_FREE(dest);
     return 0;
 }
+
+void
+compress_cleanup(struct ssh_crypto_struct *crypto)
+{
+    if (crypto->compress_out_ctx) {
+        deflateEnd(crypto->compress_out_ctx);
+    }
+    SAFE_FREE(crypto->compress_out_ctx);
+
+    if (crypto->compress_in_ctx) {
+        inflateEnd(crypto->compress_in_ctx);
+    }
+    SAFE_FREE(crypto->compress_in_ctx);
+}
diff --git a/src/wrapper.c b/src/wrapper.c
index 32f2877f..e3a71622 100644
--- a/src/wrapper.c
+++ b/src/wrapper.c
@@ -38,10 +38,6 @@
 #include <stdio.h>
 #include <string.h>
 
-#ifdef WITH_ZLIB
-#include <zlib.h>
-#endif
-
 #include "libssh/priv.h"
 #include "libssh/session.h"
 #include "libssh/crypto.h"
@@ -202,15 +198,7 @@ void crypto_free(struct ssh_crypto_struct *crypto)
         SAFE_FREE(crypto->secret_hash);
     }
 #ifdef WITH_ZLIB
-    if (crypto->compress_out_ctx) {
-        deflateEnd(crypto->compress_out_ctx);
-    }
-    SAFE_FREE(crypto->compress_out_ctx);
-
-    if (crypto->compress_in_ctx) {
-        inflateEnd(crypto->compress_in_ctx);
-    }
-    SAFE_FREE(crypto->compress_in_ctx);
+    compress_cleanup(crypto);
 #endif /* WITH_ZLIB */
     SAFE_FREE(crypto->encryptIV);
     SAFE_FREE(crypto->decryptIV);