diff --git a/actix-http/src/ws/deflate.rs b/actix-http/src/ws/deflate.rs
index dc02f46ef..d5528d988 100644
--- a/actix-http/src/ws/deflate.rs
+++ b/actix-http/src/ws/deflate.rs
@@ -577,9 +577,8 @@ impl DeflateCompressionContext {
 
 #[cfg(test)]
 mod tests {
-    use crate::body::MessageBody;
-
     use super::*;
+    use crate::body::MessageBody;
 
     #[test]
     fn test_session_parameters() {
diff --git a/awc/src/ws.rs b/awc/src/ws.rs
index 376dccaa9..507bcd3c4 100644
--- a/awc/src/ws.rs
+++ b/awc/src/ws.rs
@@ -450,8 +450,9 @@ impl WebsocketsRequest {
 
             framed.into_map_codec(move |_| {
                 let codec = if let Some(parameter) = selected_parameter.clone() {
-                    let context = parameter.create_context(self.deflate_compression_level, false);
-                    Codec::new_deflate(context)
+                    let (compress, decompress) =
+                        parameter.create_context(self.deflate_compression_level, false);
+                    Codec::new_deflate(compress, decompress)
                 } else {
                     Codec::new()
                 }