diff --git a/src/bindings/swig/hammer.i b/src/bindings/swig/hammer.i
index 357018b3d3663f44fbf4935da5687ff00a977267..ac8c47f4c16d7899bc0f7557b73fd17d83788acc 100644
--- a/src/bindings/swig/hammer.i
+++ b/src/bindings/swig/hammer.i
@@ -1,5 +1,11 @@
 %module hammer
 
-%import "allocator.h"
-%import "hammer.h"
+%{
+#include "allocator.h"
+#include "hammer.h"
+#include "internal.h"
+%}
+%include "allocator.h"
+%include "hammer.h"
+
 
diff --git a/src/hammer.h b/src/hammer.h
index c8a2bf380a91df3ba1bde041151e1f68dcde24a8..2aded7a2c6ecf1e13f3f368b4953ee004a1475d7 100644
--- a/src/hammer.h
+++ b/src/hammer.h
@@ -660,7 +660,7 @@ HParsedToken *h_act_ignore(const HParseResult *p, void* userdata);
 // {{{ Benchmark functions
 HAMMER_FN_DECL(HBenchmarkResults *, h_benchmark, HParser* parser, HParserTestcase* testcases);
 void h_benchmark_report(FILE* stream, HBenchmarkResults* results);
-void h_benchmark_dump_optimized_code(FILE* stream, HBenchmarkResults* results);
+//void h_benchmark_dump_optimized_code(FILE* stream, HBenchmarkResults* results);
 // }}}
 
 // {{{ Token type registry