diff --git a/tests/test_merkle_tree.cairo b/tests/test_merkle_tree.cairo
index a1cac5b..dd63f95 100644
--- a/tests/test_merkle_tree.cairo
+++ b/tests/test_merkle_tree.cairo
@@ -335,7 +335,10 @@ fn test_john_claims_multiple_allocations() {
     assert!(contract.check_claimed(john_address, timestamp2, amount2));
     assert!(!contract.check_claimed(john_address, timestamp3, amount3));
 
+    start_cheat_caller_address(offsetter_address, owner_address);
     contract.set_merkle_root(new_root);
+
+    start_cheat_caller_address(offsetter_address, john_address);
     contract.claim(amount4, timestamp4, proof4);
     assert!(contract.check_claimed(john_address, timestamp4, amount4));
     assert!(!contract.check_claimed(john_address, timestamp3, amount3));