diff --git a/rasa-ext-style.cabal b/rasa-ext-style.cabal
--- a/rasa-ext-style.cabal
+++ b/rasa-ext-style.cabal
@@ -1,5 +1,5 @@
 name: rasa-ext-style
-version: 0.1.1
+version: 0.1.2
 cabal-version: >=1.10
 build-type: Simple
 license: MIT
@@ -22,7 +22,7 @@
         Rasa.Ext.Style
     build-depends:
         base >=4.8 && <5,
-        rasa >=0.1.4 && <0.2,
+        rasa >=0.1.6 && <0.2,
         lens ==4.14.*,
         data-default >=0.7.1.1 && <0.8
     default-language: Haskell2010
diff --git a/src/Rasa/Ext/Style.hs b/src/Rasa/Ext/Style.hs
--- a/src/Rasa/Ext/Style.hs
+++ b/src/Rasa/Ext/Style.hs
@@ -6,6 +6,7 @@
 
 import Data.Default
 import Control.Applicative
+import Control.Monad
 
 -- | These represent the possible colors for 'fg' or 'bg'.
 -- 'DefColor' represents the terminal's default color.
@@ -57,7 +58,7 @@
 makeLenses ''Styles
 
 -- | A lens over the styles stored in the current buffer.
-styles :: Lens' Buffer [Span Style]
+styles :: HasBuffer s => Lens' s [Span Style]
 styles = bufExt.styles'
 
 instance Default Styles where
@@ -86,5 +87,5 @@
 -- > rasa $ do
 -- >    style
 -- >    ...
-style :: Scheduler ()
-style = afterRender $ buffersDo_ $ styles .= []
+style :: Action ()
+style = void . afterRender $ buffersDo_ $ styles .= []