diff --git a/tests/safe-integration/test_guard_safe_e2e.py b/tests/safe-integration/test_guard_safe_e2e.py
index fc323afd..7cd8791e 100644
--- a/tests/safe-integration/test_guard_safe_e2e.py
+++ b/tests/safe-integration/test_guard_safe_e2e.py
@@ -378,4 +378,3 @@ def test_swap_through_module_unauthorised(
     with pytest.raises(ValueError) as e:
         ts_module.functions.performCall(target, call_data).transact({"from": attacker_account})
     assert "validateCall: Sender not allowed" in str(e)
-
diff --git a/tests/safe-integration/test_guard_safe_uniswap_v2.py b/tests/safe-integration/test_guard_safe_uniswap_v2.py
index 0c488cd5..5bf06f3b 100644
--- a/tests/safe-integration/test_guard_safe_uniswap_v2.py
+++ b/tests/safe-integration/test_guard_safe_uniswap_v2.py
@@ -219,8 +219,7 @@ def test_safe_module_initialised(
     """Vault and guard are initialised for the owner."""
     assert guard.functions.owner().call() == owner
     assert guard.functions.isAllowedSender(asset_manager).call() is True
-    assert guard.functions.isAllowedWithdrawDestination(owner).call() is True
-    assert guard.functions.isAllowedWithdrawDestination(asset_manager).call() is False
+    assert guard.functions.isAllowedSender(safe.address).call() is False
 
     # We have accessed needed for a swap
     assert guard.functions.callSiteCount().call() == 5