diff --git a/src/main/wacc/parser.scala b/src/main/wacc/parser.scala
index 2f41ab0..7da0a0a 100644
--- a/src/main/wacc/parser.scala
+++ b/src/main/wacc/parser.scala
@@ -126,7 +126,7 @@ object parser {
         (`<type>`.label("function declaration") <~> `<ident>` <~ "(")
       ) <**> `<partial-func-decl>`
     ).label("function declaration"),
-    ((`<ident>` <~ "(") *> fail("function is missing return type") | `<stmt>`.label(
+    (atomic(`<ident>` <~ "(") ~> fail("function is missing return type") | `<stmt>`.label(
       "main program body"
     )) <~ "end"
   )