1 files changed, 7 insertions, 2 deletions
diff --git a/src/main.c b/src/main.c
index 0e6807d..5827913 100644
--- a/src/main.c
+++ b/src/main.c
@@ -1,3 +1,6 @@
+#include <stdio.h>
+#include <stdlib.h>
+
 #include <context.h>
 #include <lint.h>
 #include <log.h>
@@ -7,8 +10,10 @@
 
 int main(int argc, char *argv[])
 {
-	if (argc < 2)
-		err("Not enough arguments!");
+	if (argc < 2) {
+		fprintf(stderr, "Not enough arguments!");
+		exit(1);
+	}
 
 	struct ctx *ctx = context_create(argv[1]);
 	preprocess(ctx);