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 a7046d0adb61cb8a3de975f989dd8df25ed48799..d8c613dfa0c6944420b1ca56b92f89801cbb3be3 100644
--- a/src/hammer.h
+++ b/src/hammer.h
@@ -660,7 +660,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);
 // }}}
 
 #endif // #ifndef HAMMER_HAMMER__H