Fix preview_session

prismicio-community · Nov 13, 2014 · 0227cf8 · 0227cf8
1 parent 2d89bfe
commit 0227cf8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/prismic/api.py b/prismic/api.py
@@ -128,7 +128,7 @@ def preview_session(self, token, link_resolver, default_url):
         main_document_id = _get_json(token).get("mainDocument")
         if main_document_id is None:
             return default_url
-        response = self.form("everything").ref(token).query(predicates.at("document.id")).submit()
+        response = self.form("everything").ref(token).query(predicates.at("document.id", main_document_id)).submit()
         if len(response.results) == 0:
             return default_url
         return link_resolver(response.results[0].as_link())