diff --git a/examples/server/json-schema-to-grammar.cpp b/examples/server/json-schema-to-grammar.cpp
index 31f321b5d..163a240bd 100644
--- a/examples/server/json-schema-to-grammar.cpp
+++ b/examples/server/json-schema-to-grammar.cpp
@@ -667,7 +667,6 @@ public:
     }
 };
 
-
 string json_schema_to_grammar(const json& schema) {
   SchemaConverter converter(/* fetch_json= */ std::nullopt, /* dotall= */ false);
   auto copy = schema;
@@ -676,23 +675,3 @@ string json_schema_to_grammar(const json& schema) {
   converter.check_errors();
   return converter.format_grammar();
 }
-
-#ifdef LLAMA_BUILD_JSON_SCHEMA_CONVERTER
-
-int main(int argc, const char** argv) {
-    if (argc != 2) {
-        cerr << "Expected only one argument" << endl;
-        return -1;
-    }
-    string file(argv[1]);
-    string schema;
-    if (file == "-") {
-        schema.append(istreambuf_iterator<char>(cin), istreambuf_iterator<char>());
-    } else {
-        ifstream in(argv[1]);
-        schema.append(istreambuf_iterator<char>(in), istreambuf_iterator<char>());
-    }
-    cout << json_schema_to_grammar(json::parse(schema)).c_str() << endl;
-}
-
-#endif
diff --git a/tests/test-json-schema-to-grammar.cpp b/tests/test-json-schema-to-grammar.cpp
index 07b284f1b..3be705fe7 100755
--- a/tests/test-json-schema-to-grammar.cpp
+++ b/tests/test-json-schema-to-grammar.cpp
@@ -2,17 +2,10 @@
 #undef NDEBUG
 #endif
 
-#include "llama.h"
-#include "grammar-parser.h"
-#include <cstdio>
-#include <cstdlib>
-#include <filesystem>
+#include "../examples/server/json-schema-to-grammar.h"
 #include <iostream>
-#include <istream>
 #include <fstream>
 #include <sstream>
-#include "../examples/server/json-schema-to-grammar.h"
-#include <cassert>
 #include <regex>
 
 using namespace std;