2 files changed, 20 insertions, 5 deletions
diff --git a/zkvms/nexus/host/src/main.rs b/zkvms/nexus/host/src/main.rs
index 68593fc..95eafe2 100644
--- a/zkvms/nexus/host/src/main.rs
+++ b/zkvms/nexus/host/src/main.rs
@@ -1,8 +1,7 @@
 use nexus_sdk::{stwo::seq::Stwo, Local, Prover, Verifiable, Viewable};
 use zkvms_host_io::{
-    benchmarkable, read_args, Input, Return,
+    benchmarkable, output_proof_size, read_args, Input, Return,
     RunType::{Execute, Prove, Verify},
-    output_proof_size,
 };
 
 fn main() {
diff --git a/zkvms/nexus/wrapper_macro/src/lib.rs b/zkvms/nexus/wrapper_macro/src/lib.rs
index 69d1de2..aa072fd 100644
--- a/zkvms/nexus/wrapper_macro/src/lib.rs
+++ b/zkvms/nexus/wrapper_macro/src/lib.rs
@@ -35,13 +35,29 @@ pub fn make_wrapper(item: TokenStream) -> TokenStream {
     let fd = FunctionDefinition::new(&item);
 
     let mut out = TokenStream::new();
-    out.extend(format!("let ({}) = read_public_input::<({})>().unwrap();", fd.grouped_public_patterns(), fd.grouped_public_types()).parse::<TokenStream>());
-    out.extend(format!("let ({}) = read_private_input::<({})>().unwrap();", fd.grouped_private_patterns(), fd.grouped_private_types()).parse::<TokenStream>());
+    out.extend(
+        format!(
+            "let ({}) = read_public_input::<({})>().unwrap();",
+            fd.grouped_public_patterns(),
+            fd.grouped_public_types()
+        )
+        .parse::<TokenStream>(),
+    );
+    out.extend(
+        format!(
+            "let ({}) = read_private_input::<({})>().unwrap();",
+            fd.grouped_private_patterns(),
+            fd.grouped_private_types()
+        )
+        .parse::<TokenStream>(),
+    );
 
     out.extend(
         format!(
             "write_public_output::<{}>(&zkp::{}({}));",
-            fd.return_type, fd.name, fd.grouped_patterns()
+            fd.return_type,
+            fd.name,
+            fd.grouped_patterns()
         )
         .parse::<TokenStream>(),
     );