diff --git a/CCTF_Solutions_main/hash.zok b/CCTF_Solutions_main/hash.zok
index d80a46e..ee842b5 100644
--- a/CCTF_Solutions_main/hash.zok
+++ b/CCTF_Solutions_main/hash.zok
@@ -1,11 +1,10 @@
 import "hashes/sha256/sha256Padded.zok" as sha256;
 from "utils/casts.zok" import cast;
 
-def main(public u32[5] address,private u8[64] flag) {
+def main(public u32[5] address,private u8[64] flag) -> u32[8] {
      u8[20] addr8 = cast(address);
-     u8[84] forHash = [...addr8,...flag];
      u32[8] genHash = sha256(flag);
      log("Hash: {} {} {} {} {} {} {} {}",genHash[0],genHash[1],genHash[2],genHash[3],genHash[4],genHash[5],genHash[6],genHash[7]);
      //assert(genHash == hash);
-     return;
+     return genHash;
 }
diff --git a/CCTF_Solutions_main/root.zok b/CCTF_Solutions_main/root.zok
index 43ac1a2..e1a4e35 100644
--- a/CCTF_Solutions_main/root.zok
+++ b/CCTF_Solutions_main/root.zok
@@ -3,9 +3,8 @@ from "utils/casts.zok" import cast;
 
 def main(public u32[8] hash,public u32[5] address,private u8[64] flag) {
      u8[20] addr8 = cast(address);
-     u8[84] forHash = [...addr8,...flag];
      u32[8] genHash = sha256(flag);
      log("Hash: {} {} {} {} {} {} {} {}",genHash[0],genHash[1],genHash[2],genHash[3],genHash[4],genHash[5],genHash[6],genHash[7]);
      assert(genHash == hash);
-     return;
+     return sha256([...addr8,...sha256(flag)]);
 }
\ No newline at end of file