Fix serving the variant pages

wagtail · May 31, 2017 · 8465e6d · 8465e6d
1 parent cf81011
commit 8465e6d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/wagtail_personalisation/wagtail_hooks.py b/src/wagtail_personalisation/wagtail_hooks.py
@@ -80,10 +80,11 @@ def serve_variation(page, request, serve_args, serve_kwargs):
 
     if user_segments:
         metadata = page.personalisable_metadata
+
+        # TODO: This is never more then one page? (fix query count)
         variations = metadata.variants_for_segments(user_segments)
         if variations:
-            variation = variations.first()
-            impersonate_other_page(variation, page)
+            variation = variations.first().variant.specific
             return variation.serve(request, *serve_args, **serve_kwargs)