From 78b20f3603ea35f30e830206d58e470bed49c5bf Mon Sep 17 00:00:00 2001
From: Romain Paquet <rpqt@rpqt.fr>
Date: Fri, 6 Dec 2024 18:28:37 +0100
Subject: [PATCH] typing: cleanup imports and fix unit in return

---
 src/typing/mod.rs | 15 +++++----------
 1 file changed, 5 insertions(+), 10 deletions(-)

diff --git a/src/typing/mod.rs b/src/typing/mod.rs
index 5fce74a..3023ae7 100644
--- a/src/typing/mod.rs
+++ b/src/typing/mod.rs
@@ -1,15 +1,10 @@
+mod error;
+
+use crate::ast::*;
+use crate::typing::error::{TypeAndSpan, TypeError, TypeErrorKind};
 use std::collections::HashMap;
 use std::fmt::Display;
 
-use BinaryExpression;
-use ReturnStatement;
-
-use crate::ast::ModulePath;
-use crate::ast::*;
-
-mod error;
-use crate::typing::error::{TypeAndSpan, TypeError, TypeErrorKind};
-
 #[cfg(test)]
 mod tests;
 
@@ -178,7 +173,7 @@ impl TypeCheck for FunctionDefinition {
             );
         }
 
-        Ok(self.return_type.clone().unwrap())
+        Ok(self.return_type.clone().unwrap_or(Type::Unit))
     }
 }