diff --git a/src/wagtail_personalisation/wagtail_hooks.py b/src/wagtail_personalisation/wagtail_hooks.py
index 34f7439..47843c3 100644
--- a/src/wagtail_personalisation/wagtail_hooks.py
+++ b/src/wagtail_personalisation/wagtail_hooks.py
@@ -141,14 +141,13 @@ def _check_for_variations(segments, page):
     """
     for segment in segments:
         page_class = page.__class__
-        if not any(item == PersonalisablePage for item in page_class.__bases__):
-            page_class = PersonalisablePage
+        if any(item == PersonalisablePage for item in page_class.__bases__):
 
-        variation = page_class.objects.filter(
-            canonical_page=page, segment=segment)
+            variation = page_class.objects.filter(
+                canonical_page=page, segment=segment)
 
-        if variation:
-            return variation
+            if variation:
+                return variation
 
     return None