1 files changed, 3 insertions, 4 deletions
diff --git a/source/tests/main.test.c b/source/tests/main.test.c
index 790c8c2..d81a8cb 100644
--- a/source/tests/main.test.c
+++ b/source/tests/main.test.c
@@ -1,10 +1,9 @@
-#include "../kit/status.h"
-#include "../kit_test/bench.h"
-#include "../kit_test/test.h"
+#define KIT_TEST_IMPLEMENTATION
+#include "../kit/kit_test.h"
 
 int main(int argc, char **argv) {
   int status = run_tests(argc, argv);
-  if (status == KIT_OK)
+  if (status == 0)
     status = run_benchmarks(argc, argv);
   return status;
 }