diff --git a/lang/lex.c b/lang/lex.c
index 3aaf2de..15ef983 100644
--- a/lang/lex.c
+++ b/lang/lex.c
@@ -1,15 +1,17 @@
+#include "lex.h"
+
 #include <blue/core/hash.h>
 #include <blue/core/queue.h>
-#include <blue/object/string.h>
 #include <blue/object/dict.h>
 #include <blue/object/number.h>
+#include <blue/object/string.h>
 #include <ctype.h>
 #include <ivy/lang/lex.h>
 #include <stdbool.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
-#include "lex.h"
+#include <wctype.h>
 
 #define LINEBUF_DEFAULT_CAPACITY 1024
 
@@ -410,7 +412,7 @@ static int advance(struct ivy_lexer *lex)
 	return c;
 }
 
-static bool input_available(struct ivy_lexer* lex)
+static bool input_available(struct ivy_lexer *lex)
 {
 	return lex->lex_linebuf_ptr < lex->lex_linebuf_len;
 }
diff --git a/lang/lex.h b/lang/lex.h
index c48d717..c95f88c 100644
--- a/lang/lex.h
+++ b/lang/lex.h
@@ -2,8 +2,10 @@
 #define _LEX_H_
 
 #include <blue/core/queue.h>
-#include <blue/object/string.h>
 #include <blue/object/dict.h>
+#include <blue/object/string.h>
+#include <ivy/lang/lex.h>
+#include <ivy/status.h>
 #include <stdint.h>
 
 struct ivy_lexer {
@@ -52,4 +54,4 @@ struct lex_token_def {
 	uint64_t name_hash;
 };
 
-#endif
\ No newline at end of file
+#endif