1 files changed, 8 insertions, 2 deletions
diff --git a/main.c b/main.c
index 8f1b25e..9e388a3 100644
--- a/main.c
+++ b/main.c
@@ -17,12 +17,18 @@ int main(int argc, char **argv) {
 #ifdef TOC_DEBUG
 	test_all();
 #endif
+
+	const char *in_filename;
 	if (argc < 2) {
+#ifdef TOC_DEBUG
+		in_filename = "test.toc";
+#else
 		fprintf(stderr, "Please specify an input file.\n");
 		return EXIT_FAILURE;
+#endif
+	} else {
+		in_filename = argv[1];
 	}
-
-	const char *in_filename = argv[1];
 	const char *out_filename = "out.c";
 
 	for (int i = 2; i < argc-1; i++) {