Apply do-syntax.

1 files changed, 2 insertions, 1 deletions

diff --git a/monkeypatch.hs b/monkeypatch.hs
index b72e346..37e52a6 100644
--- a/monkeypatch.hs
+++ b/monkeypatch.hs
@@ -49,6 +49,7 @@ import Text.PrettyPrint             (Doc, doubleQuotes, empty, text, vcat, ($$),
                                      (<+>))
 import Text.Show.Pretty
 
+import Sweeten
 import GrepNested
 
 trace :: p -> a -> a
@@ -631,7 +632,7 @@ transpile o fname incs (CTranslUnit edecls _) = do
                                 return $ foldr applyComputation (Computation Map.empty Map.empty retUnit) xs
                         case mhask of
                             Just hask -> do printHeader
-                                            mapM_ (putStrLn . ("  "++)) $ lines $ HS.prettyPrint $ comp hask
+                                            mapM_ (putStrLn . ("  "++)) $ lines $ HS.prettyPrint $ applyDoSyntax $ comp hask
                             Nothing -> forM_ (oSelectFunction o) $ \_ -> do
                                 printHeader
                                 forM_ bdy $ \d -> do