Unexpected return error

2 files changed, 4 insertions, 1 deletions

diff --git a/src/Compiler.hs b/src/Compiler.hs
index da25c5d..413a041 100644
--- a/src/Compiler.hs
+++ b/src/Compiler.hs
@@ -121,7 +121,7 @@ compile x = do
     (A.Return value pos) -> do
       (ev, errs) <- get
       case getSyml ev "$fn$" of
-        Nothing -> addError $ Error "return without function call" pos
+        Nothing -> addError $ UnexpectedReturn "return without function call" pos
         Just (_, A.FuncType _ rtyp, _) -> do
           r <- typecheckReturn value rtyp
           case r of
diff --git a/src/Error.hs b/src/Error.hs
index b528f9e..1363120 100644
--- a/src/Error.hs
+++ b/src/Error.hs
@@ -6,6 +6,7 @@ import Text.Parsec.Error (ParseError, errorMessages, showErrorMessages)
 
 data Error
   = TypeError String SourcePos
+  | UnexpectedReturn String SourcePos
   | Error String SourcePos
   deriving (Eq)
 
@@ -15,6 +16,8 @@ instance Show Error where
   -- XXX: can we do this differently?
   show (TypeError message pos) =
     show pos ++ " error: " ++ message
+  show (UnexpectedReturn message pos) =
+    show pos ++ " error: " ++ message
 
 showParserError :: ParseError -> String
 showParserError error =