some updates.

[hs-java.git] / JVM / Generator.hs

diff --git a/JVM/Generator.hs b/JVM/Generator.hs
index 2ee0240f00726dc24049ffc6c862e54dd0ae1fbd..34b27edb76de907fcfd7f827cde0d90c8eeb24a5 100644 (file)
--- a/JVM/Generator.hs
+++ b/JVM/Generator.hs
@@ -134,11 +134,13 @@ endMethod = do
                    currentMethod = Nothing,
                    doneMethods = doneMethods st ++ [method']}
 
-newMethod :: [AccessFlag] -> B.ByteString -> [ArgumentSignature] -> ReturnSignature -> Generate () -> Generate ()
+newMethod :: [AccessFlag] -> B.ByteString -> [ArgumentSignature] -> ReturnSignature -> Generate () -> Generate (NameType Method)
 newMethod flags name args ret gen = do
-  startMethod flags name (MethodSignature args ret)
+  let sig = MethodSignature args ret
+  startMethod flags name sig
   gen
   endMethod
+  return (NameType name sig)
 
 genCode :: GState -> Code
 genCode st = Code {