1 files changed, 25 insertions, 3 deletions
diff --git a/zkvms/zkm/wrapper_macro/src/lib.rs b/zkvms/zkm/wrapper_macro/src/lib.rs
index 6a88602..fde5fa3 100644
--- a/zkvms/zkm/wrapper_macro/src/lib.rs
+++ b/zkvms/zkm/wrapper_macro/src/lib.rs
@@ -33,10 +33,32 @@ pub fn make_wrapper(item: TokenStream) -> TokenStream {
 
     let mut out = TokenStream::new();
     // NOTE: The first read returns public data, the second returns private
-    out.extend(format!("let ({}) : ({}) = read();", fd.grouped_public_patterns(), fd.grouped_public_types()).parse::<TokenStream>());
-    out.extend(format!("let ({}) : ({}) = read();", fd.grouped_private_patterns(), fd.grouped_private_types()).parse::<TokenStream>());
+    out.extend(
+        format!(
+            "let ({}) : ({}) = read();",
+            fd.grouped_public_patterns(),
+            fd.grouped_public_types()
+        )
+        .parse::<TokenStream>(),
+    );
+    out.extend(
+        format!(
+            "let ({}) : ({}) = read();",
+            fd.grouped_private_patterns(),
+            fd.grouped_private_types()
+        )
+        .parse::<TokenStream>(),
+    );
 
-    out.extend(format!("commit::<{}>(&zkp::{}({}));", fd.return_type, fd.name, fd.grouped_patterns()).parse::<TokenStream>());
+    out.extend(
+        format!(
+            "commit::<{}>(&zkp::{}({}));",
+            fd.return_type,
+            fd.name,
+            fd.grouped_patterns()
+        )
+        .parse::<TokenStream>(),
+    );
 
     let mut block = TokenStream::new();
     block.extend(format!("{{ {} }}", out).parse::<TokenStream>());