Support both IO and clean version of Generate monad.

[hs-java.git] / TestGen.hs

diff --git a/TestGen.hs b/TestGen.hs
index f21912c58e863481124923c72bbb70eddac2dea5..31e71c0676ae0255d345a46d8ac1d6a502dd1dd5 100644 (file)
--- a/TestGen.hs
+++ b/TestGen.hs
@@ -11,7 +11,7 @@ import Java.ClassPath
 import qualified Java.Lang
 import qualified Java.IO
 
-test :: Generate ()
+test :: GenerateIO ()
 test = do
   withClassPath $ do
       -- Add current directory (with Hello.class) to ClassPath
@@ -63,6 +63,6 @@ test = do
 
 main :: IO ()
 main = do
-  testClass <- generate [] "Test" test
+  testClass <- generateIO [] "Test" test
   B.writeFile "Test.class" (encodeClass testClass)