diff --git a/Cargo.lock b/Cargo.lock
index 3b65cb0..3aa0214 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -519,11 +519,13 @@ version = "0.1.0"
 dependencies = [
  "base64",
  "chrono",
+ "prettyplease",
  "progenitor",
  "progenitor-client",
  "reqwest",
  "serde",
  "serde_json",
+ "syn 1.0.109",
  "uuid",
  "wasm-bindgen-test",
 ]
diff --git a/example-wasm/Cargo.toml b/example-wasm/Cargo.toml
index ee40647..904f476 100644
--- a/example-wasm/Cargo.toml
+++ b/example-wasm/Cargo.toml
@@ -15,5 +15,7 @@ uuid = { version = "1.3", features = ["serde", "v4", "js"] }
 wasm-bindgen-test = "0.3.34"
 
 [build-dependencies]
+prettyplease = "0.1.25"
 progenitor = { path = "../progenitor" }
 serde_json = "1.0"
+syn = "1.0"
diff --git a/example-wasm/build.rs b/example-wasm/build.rs
index ffb0ab4..2119be5 100644
--- a/example-wasm/build.rs
+++ b/example-wasm/build.rs
@@ -13,7 +13,9 @@ fn main() {
     let spec = serde_json::from_reader(file).unwrap();
     let mut generator = progenitor::Generator::default();
 
-    let content = generator.generate_text(&spec).unwrap();
+    let tokens = generator.generate_tokens(&spec).unwrap();
+    let ast = syn::parse2(tokens).unwrap();
+    let content = prettyplease::unparse(&ast);
 
     let mut out_file = Path::new(&env::var("OUT_DIR").unwrap()).to_path_buf();
     out_file.push("codegen.rs");