1 files changed, 7 insertions, 7 deletions

diff --git a/src/main.c b/src/main.c
index 4f68256..ca6ed9b 100644
--- a/src/main.c
+++ b/src/main.c
@@ -27,11 +27,11 @@
 #include "string_view.h"
 
 static void
-generate_gas_x86_64_linux(ast_node_t *func)
+generate_gas_x86_64_linux(ast_node_t *ns)
 {
   gas_assembly_generator_t gen;
   gas_assembly_generator_init(&gen, stdout);
-  gas_assembly_generator_compile(&gen, func);
+  gas_assembly_generator_compile_linux_main(&gen, ns);
 }
 
 static void
@@ -89,20 +89,20 @@ main(int argc, char **argv)
   parser_t parser;
   parser_init(&parser, &lexer, scope);
 
-  ast_node_t *func = parser_parse_function_declaration(&parser);
+  ast_node_t *ns = parser_parse_ns(&parser);
 
-  if (func == NULL) {
+  if (ns == NULL) {
     parser_print_errors(&parser);
     return EXIT_FAILURE;
   }
 
   if (should_dump_ast) {
-    pretty_print_ast(func);
+    pretty_print_ast(ns);
   } else {
-    generate_gas_x86_64_linux(func);
+    generate_gas_x86_64_linux(ns);
   }
 
   scope_destroy(scope);
-  ast_node_destroy(func);
+  ast_node_destroy(ns);
   return EXIT_SUCCESS;
 }