From ba0246a2ec9c447447de51d4745b6eb3fe207db5 Mon Sep 17 00:00:00 2001
From: Parbhat Puri <parbhatpuri17@gmail.com>
Date: Tue, 10 Nov 2020 16:36:59 +0530
Subject: [PATCH] Add snippetChooser URL as removed from Wagtail

---
 wagtail_draftail_snippet/wagtail_hooks.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/wagtail_draftail_snippet/wagtail_hooks.py b/wagtail_draftail_snippet/wagtail_hooks.py
index 479c7a8..e56fddd 100644
--- a/wagtail_draftail_snippet/wagtail_hooks.py
+++ b/wagtail_draftail_snippet/wagtail_hooks.py
@@ -68,10 +68,12 @@ def editor_js():
     return format_html(
         """
             <script>
-                window.chooserUrls.snippetLinkModelChooser = '{0}';
-                window.chooserUrls.snippetEmbedModelChooser = '{1}';
+                window.chooserUrls.snippetChooser = '{0}';
+                window.chooserUrls.snippetLinkModelChooser = '{1}';
+                window.chooserUrls.snippetEmbedModelChooser = '{2}';
             </script>
         """,
+        reverse('wagtailsnippets:choose_generic'),
         reverse("wagtaildraftailsnippet:choose-snippet-link-model"),
         reverse("wagtaildraftailsnippet:choose-snippet-embed-model"),
     )