diff --git a/src/parser/mod.rs b/src/parser/mod.rs
index c000532..152f95c 100644
--- a/src/parser/mod.rs
+++ b/src/parser/mod.rs
@@ -75,9 +75,8 @@ impl TopLevel {
                     false
                 };
                 let name = inner.next().unwrap().as_str().to_string();
-                let params_pair = inner.next().unwrap();
-                let params = if params_pair.as_rule() == Rule::param_list {
-                    ParamList::from_pair(params_pair)
+                let params = if inner.peek().unwrap().as_rule() == Rule::param_list {
+                    ParamList::from_pair(inner.next().unwrap())
                 } else {
                     ParamList(vec![])
                 };
diff --git a/test/src/main.ms b/test/src/main.ms
index 755f17d..eaed9a5 100644
--- a/test/src/main.ms
+++ b/test/src/main.ms
@@ -15,3 +15,7 @@ function add(a: int, b: int): int {
         return 0;
     }
 }
+
+function main() {
+    add(10, 10);
+}