diff --git a/prover/examples/README.md b/prover/examples/README.md
index 2db325d9..ec062056 100644
--- a/prover/examples/README.md
+++ b/prover/examples/README.md
@@ -68,7 +68,7 @@ rustflags = ["--cfg", 'target_os="zkvm"',"-C", "target-feature=+crt-static", "-C
 * Build the Sha2/Revme
 
 ```
-cd prover/examples/sha2
+cd prover/examples/sha2-rust
 cargo build --target=mips-unknown-linux-musl
 ```
 
@@ -80,7 +80,7 @@ cargo build --target=mips-unknown-linux-musl
 
 cd ../..
 
-ARGS="711e9609339e92b03ddc0a211827dba421f38f9ed8b9d806e1ffdd8c15ffa03d world!" RUST_LOG=info ELF_PATH=examples/sha2/target/mips-unknown-linux-musl/debug/sha2-bench HOST_PROGRAM=sha2_bench SEG_OUTPUT=/tmp/output cargo run --release --example zkmips prove_host_program
+ARGS="711e9609339e92b03ddc0a211827dba421f38f9ed8b9d806e1ffdd8c15ffa03d world!" RUST_LOG=info ELF_PATH=examples/sha2-rust/target/mips-unknown-linux-musl/debug/sha2-rust HOST_PROGRAM=sha2_rust SEG_OUTPUT=/tmp/output cargo run --release --example zkmips prove_host_program
 
 Or
 
diff --git a/prover/examples/sha2-rust/Cargo.lock b/prover/examples/sha2-rust/Cargo.lock
index 4030fadb..28d0aeff 100644
--- a/prover/examples/sha2-rust/Cargo.lock
+++ b/prover/examples/sha2-rust/Cargo.lock
@@ -180,7 +180,7 @@ dependencies = [
 ]
 
 [[package]]
-name = "sha2-bench"
+name = "sha2-rust"
 version = "0.1.0"
 dependencies = [
  "sha2",
diff --git a/prover/examples/sha2-rust/rust-toolchain.toml b/prover/examples/sha2-rust/rust-toolchain.toml
index 64c5f280..f0ae574b 100644
--- a/prover/examples/sha2-rust/rust-toolchain.toml
+++ b/prover/examples/sha2-rust/rust-toolchain.toml
@@ -1,4 +1,4 @@
 [toolchain]
-channel = "nightly-2023-04-06-x86_64-unknown-linux-gnu"
+channel = "nightly-2023-04-06"
 targets = ["mips-unknown-linux-musl"]
 profile = "minimal"
diff --git a/prover/examples/zkmips.rs b/prover/examples/zkmips.rs
index d9f0a0ea..5646a7af 100644
--- a/prover/examples/zkmips.rs
+++ b/prover/examples/zkmips.rs
@@ -251,7 +251,7 @@ fn prove_sha2_rust() {
     // 1. split ELF into segs
     let elf_path = env::var("ELF_PATH").expect("ELF file is missing");
     let seg_path = env::var("SEG_OUTPUT").expect("Segment output path is missing");
-    let seg_size = env::var("SEG_SIZE").unwrap_or("0".to_string());
+    let seg_size = env::var("SEG_SIZE").unwrap_or("65536".to_string());
     let seg_size = seg_size.parse::<_>().unwrap_or(0);
 
     let mut state = load_elf_with_patch(&elf_path, vec![]);
@@ -280,7 +280,6 @@ fn prove_sha2_rust() {
     if seg_size != 0 {
         seg_num = (total_steps + seg_size - 1) / seg_size;
     }
-
     prove_multi_seg_common(&seg_path, "", "", "", seg_size, seg_num, 0).unwrap()
 }