diff --git a/apiary-websockets.cabal b/apiary-websockets.cabal
--- a/apiary-websockets.cabal
+++ b/apiary-websockets.cabal
@@ -1,5 +1,5 @@
 name:                apiary-websockets
-version:             0.4.3.4
+version:             0.7.0.0
 synopsis:            websockets supper for apiary web framework.
 description:
   example: <https://github.com/philopon/apiary/blob/master/examples/websockets.hs>
@@ -20,7 +20,7 @@
   exposed-modules:     Web.Apiary.WebSockets
   other-modules:       
   build-depends:       base               >=4.6   && <4.8
-                     , apiary             >=0.4.2 && <0.7
+                     , apiary             >=0.7   && <0.8
                      , wai-websockets     >=2.1   && <2.2
                      , websockets         >=0.8   && <0.9
 
diff --git a/src/Web/Apiary/WebSockets.hs b/src/Web/Apiary/WebSockets.hs
--- a/src/Web/Apiary/WebSockets.hs
+++ b/src/Web/Apiary/WebSockets.hs
@@ -24,10 +24,10 @@
     , sendTextData, sendBinaryData, sendClose, sendPing
     )
 
-actionWithWebSockets' :: Monad m => WS.ConnectionOptions 
+actionWithWebSockets' :: WS.ConnectionOptions 
                       -> Fn xs WS.ServerApp
-                      -> Fn xs (ActionT m ())
-                      -> ApiaryT xs m ()
+                      -> Fn xs (Action ())
+                      -> Apiary xs ()
 actionWithWebSockets' conn srv m = do
     actionWithPreAction pa m
   where
@@ -37,8 +37,7 @@
             Nothing   -> return ()
             Just resp -> stopWith resp
 
-actionWithWebSockets :: Monad m 
-                     => Fn c WS.ServerApp
-                     -> Fn c (ActionT m ())
-                     -> ApiaryT c m ()
+actionWithWebSockets :: Fn c WS.ServerApp
+                     -> Fn c (Action ())
+                     -> Apiary c ()
 actionWithWebSockets = actionWithWebSockets' WS.defaultConnectionOptions