diff --git a/cfgov/v1/wagtail_hooks.py b/cfgov/v1/wagtail_hooks.py index 4be35035577..34ceb70e6c2 100644 --- a/cfgov/v1/wagtail_hooks.py +++ b/cfgov/v1/wagtail_hooks.py @@ -61,17 +61,20 @@ def get_url(self, context): @hooks.register("construct_page_action_menu") def add_language_links(menu_items, request, context): - page = context["page"] - return menu_items.extend( - [ - LanguageMenuItem( - f"Edit {languages[translation.language]} page", - f"/admin/pages/{translation.pk}/edit/", - ) - for translation in page.get_translations() - if translation.language != page.language - ] - ) + try: + page = context["page"] + return menu_items.extend( + [ + LanguageMenuItem( + f"Edit {languages[translation.language]} page", + f"/admin/pages/{translation.pk}/edit/", + ) + for translation in page.get_translations() + if translation.language != page.language + ] + ) + except KeyError: + pass @hooks.register("after_delete_page")