diff --git a/package.json b/package.json
index 9517ba1..c0f027a 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "merkle-tree-zilliqa-js",
-  "version": "1.0.2",
+  "version": "1.0.4",
   "description": "A JS Library to generate the merkle tree and gets it's root and proof for any given distribution",
   "main": "lib/index.js",
   "types": "lib/index.d.ts",
diff --git a/src/tree-generator.ts b/src/tree-generator.ts
index b3759c0..fa511ca 100644
--- a/src/tree-generator.ts
+++ b/src/tree-generator.ts
@@ -34,7 +34,7 @@ export const generateTreeforBech32 = (props: DataInterface[]) => {
     });
   });
   const tree = new MerkleTree(leavesData, sha256, {
-    sortLeaves: true,
+    sort: true,
   });
   const merkleRoot = tree.getRoot().toString('hex');
   accumulator.forEach((data) => {
@@ -57,7 +57,7 @@ export const generateTreeforBase16 = (props: DataInterface[]) => {
     });
   });
   const tree = new MerkleTree(leavesData, sha256, {
-    sortLeaves: true,
+    sort: true,
   });
   const merkleRoot = tree.getRoot().toString('hex');
   accumulator.forEach((data) => {