diff --git a/push-notify-apn.cabal b/push-notify-apn.cabal
index 11f78f4..885d81f 100644
--- a/push-notify-apn.cabal
+++ b/push-notify-apn.cabal
@@ -30,7 +30,7 @@ library
                      , containers
                      , data-default
                      , http2
-                     , http2-client
+                     , http2-client >= 0.10.0.1
                      , lifted-base
                      , mtl
                      , random
diff --git a/src/Network/PushNotify/APN.hs b/src/Network/PushNotify/APN.hs
index c44febc..022a220 100644
--- a/src/Network/PushNotify/APN.hs
+++ b/src/Network/PushNotify/APN.hs
@@ -639,7 +639,7 @@ sendApnRaw connection deviceToken mJwtBearerToken message = bracket_
                 upload message (HTTP2.setEndHeader . HTTP2.setEndStream) client (_outgoingFlowControl client) stream osfc
                 let pph _hStreamId _hStream hHeaders _hIfc _hOfc =
                         lift $ print hHeaders
-                response <- waitStream stream isfc pph
+                response <- waitStream client stream isfc pph
                 let (errOrHeaders, frameResponses, _) = response
                 case errOrHeaders of
                     Left err -> throwIO (ApnExceptionHTTP $ toErrorCodeId err)