diff --git a/src/wagtail_personalisation/wagtail_hooks.py b/src/wagtail_personalisation/wagtail_hooks.py
index e3600824..d29cc23b 100644
--- a/src/wagtail_personalisation/wagtail_hooks.py
+++ b/src/wagtail_personalisation/wagtail_hooks.py
@@ -132,7 +132,7 @@ def page_listing_variant_buttons(page, *args, **kwargs):
 
     """
     if len(args) == 0 and kwargs.get("user"):
-        page_perms = kwargs.get("user").permission_set_for_page(page)
+        page_perms = page.permissions_for_user(kwargs["user"])
     elif len(args) == 1:
         page_perms = args[0]
     else: