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"),
     )