diff --git a/wagtail/documents/wagtail_hooks.py b/wagtail/documents/wagtail_hooks.py
index ee27c3ca30..3911b85beb 100644
--- a/wagtail/documents/wagtail_hooks.py
+++ b/wagtail/documents/wagtail_hooks.py
@@ -1,9 +1,8 @@
 from django.conf import settings
 from django.conf.urls import include, url
-from django.contrib.staticfiles.templatetags.staticfiles import static
 from django.template.response import TemplateResponse
 from django.urls import reverse
-from django.utils.html import format_html, format_html_join
+from django.utils.html import format_html
 from django.utils.translation import ugettext_lazy as _
 from django.utils.translation import ugettext, ungettext
 
@@ -57,14 +56,7 @@ def register_documents_menu_item():
 
 @hooks.register('insert_editor_js')
 def editor_js():
-    js_files = [
-        static('wagtaildocs/js/document-chooser.js'),
-    ]
-    js_includes = format_html_join(
-        '\n', '',
-        ((filename, ) for filename in js_files)
-    )
-    return js_includes + format_html(
+    return format_html(
         """
         ',
-        ((filename, ) for filename in js_files)
-    )
-    return js_includes + format_html(
+    return format_html(
         """
         
-            
+            
         """,
-        static('wagtailsnippets/js/snippet-chooser.js'),
         reverse('wagtailsnippets:choose_generic')
     )
 
diff --git a/wagtail/snippets/widgets.py b/wagtail/snippets/widgets.py
index 3345887954..c7f0b26746 100644
--- a/wagtail/snippets/widgets.py
+++ b/wagtail/snippets/widgets.py
@@ -39,3 +39,6 @@ class AdminSnippetChooser(AdminChooser):
             model=json.dumps('{app}/{model}'.format(
                 app=model._meta.app_label,
                 model=model._meta.model_name)))
+
+    class Media:
+        js = ['wagtailsnippets/js/snippet-chooser.js']