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 5dac8bf84ec45cda9c66b46ff68dfb2be374f89b..b4fc823dfc22bdb3ab6205e2071f556931c9ad3e 100644
--- a/src/hammer.h
+++ b/src/hammer.h
@@ -666,7 +666,7 @@ HParsedToken *h_act_ignore(const HParseResult *p);
 // {{{ 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);
 // }}}
 
 #ifdef __cplusplus