updates checking for page class

2016-12-05 14:20:24 +01:00
parent 2c7544283e
commit 7071198964
1 changed files with 2 additions and 1 deletions
--- a/src/personalisation/wagtail_hooks.py
+++ b/src/personalisation/wagtail_hooks.py
@ -92,7 +92,8 @@ def serve_variation(page, request, serve_args, serve_kwargs):

 def _check_for_variations(segments, page):
    for segment in segments:
-        variation = PersonalisablePage.objects.filter(canonical_page=page, segment=segment)
+        page_class = page.__class__
+        variation = page_class.objects.filter(canonical_page=page, segment=segment)

        if variation: