1 files changed, 14 insertions, 0 deletions
diff --git a/src/output_source.cpp b/src/output_source.cpp
index 7000ef9..576ecbc 100644
--- a/src/output_source.cpp
+++ b/src/output_source.cpp
@@ -213,6 +213,13 @@ void output_source_t::write() {
 		write_line();
 	}
 
+	if (!get_generator()->get_grammar().ns.empty()) {
+		for (const std::string &ns : get_generator()->get_grammar().ns)
+			write_line("namespace ", ns, " {");
+
+		write_line();
+	}
+
 	emit_types();
 	write_line();
 
@@ -229,6 +236,13 @@ void output_source_t::write() {
 	write_line();
 
 	emit_push();
+
+	if (!get_generator()->get_grammar().ns.empty()) {
+		write_line();
+
+		for (size_t i = 0; i < get_generator()->get_grammar().ns.size(); i++)
+			write_line("}");
+	}
 }
 
 }