diff --git a/tests/user.js b/tests/user.js
index b725388..3d0b817 100644
--- a/tests/user.js
+++ b/tests/user.js
@@ -393,7 +393,7 @@ class User {
             if (singleStaking) {
                 b = 0;
             } else {
-                b = balanceStaked.mul(rwds[0]).sub(completeB).div(U64_MAX).add(pendingB).toNumber();
+                b = balanceStaked.mul(rwds[1]).sub(completeB).div(U64_MAX).add(pendingB).toNumber();
             }
             return [a, b];
             
@@ -577,4 +577,4 @@ class User {
 module.exports = {
     claimForUsers,
     User
-};
\ No newline at end of file
+};