1 files changed, 3 insertions, 1 deletions

diff --git a/src/semantic.c b/src/semantic.c
index 6c9f290..dd95d28 100644
--- a/src/semantic.c
+++ b/src/semantic.c
@@ -311,6 +311,8 @@ resolve_type(ParseTree *ast, Scope *scope, Node *node) {
             node->expr_type = type;
         } break;
         case NODE_FUN: {
+            node->expr_type = &default_types[TYPE_VOID];
+
             // Fill up new scope with parameters
             scope = alloc_scope(scope);
             array_push(ast->scopes, scope);
@@ -336,7 +338,7 @@ resolve_type(ParseTree *ast, Scope *scope, Node *node) {
                     }
                 }
                 Node *last_expr = body->block.expr[array_size(body->block.expr) - 1];
-                node->expr_type = last_expr->expr_type;
+                node->fun.body->expr_type = last_expr->expr_type;
             } else {
                 if (!resolve_type(ast, scope, body)) {
                     return false;