diff --git a/examples/simple/src/lib.rs b/examples/simple/src/lib.rs
index 12296a6aa..6c62a4634 100644
--- a/examples/simple/src/lib.rs
+++ b/examples/simple/src/lib.rs
@@ -9,14 +9,10 @@ use pulumi_wasm_rust::pulumi_main;
 
 #[pulumi_main]
 fn test_main() -> Result<(), Error> {
-    wasm_common::setup_logger();
-
     let length: Output<i32> = Output::new(&12).map(|i: i32| i * 3);
-
     let _ = create_random_string(RandomStringArgs {
         name: "test",
         length,
     });
-
     Ok(())
 }
diff --git a/pulumi_wasm_rust_macro/src/lib.rs b/pulumi_wasm_rust_macro/src/lib.rs
index 89eb696e1..e6f2fe1ad 100644
--- a/pulumi_wasm_rust_macro/src/lib.rs
+++ b/pulumi_wasm_rust_macro/src/lib.rs
@@ -14,9 +14,8 @@ pub fn pulumi_main(_attr: TokenStream, item: TokenStream) -> TokenStream {
 
     let expanded = quote! {
 
-        fn #fn_name() #fn_output {
+        fn #fn_name() #fn_output
             #fn_block
-        }
 
         #[export_name = "component:pulumi-wasm/pulumi-main@0.1.0#main"]
         unsafe extern "C" fn __exported() {