diff --git a/setup.py b/setup.py
index 1861428..01ef4fe 100644
--- a/setup.py
+++ b/setup.py
@@ -6,15 +6,13 @@
 root_dir = os.path.dirname(__file__)
 if root_dir:
     os.chdir(root_dir)
-accepted_filetypes = (".html", ".png", ".svg", ".js", ".css")
 
-for dirpath, dirnames, filenames in os.walk('ab_plugin_notebook'):
+for dirpath, dirnames, filenames in os.walk("ab_online"):
     # Ignore dirnames that start with '.'
-    if ('__init__.py' in filenames
-            or any(x.endswith(accepted_filetypes) for x in filenames)):
-        pkg = dirpath.replace(os.path.sep, '.')
+    if "__init__.py" in filenames or "jupyter_config" in dirpath:
+        pkg = dirpath.replace(os.path.sep, ".")
         if os.path.altsep:
-            pkg = pkg.replace(os.path.altsep, '.')
+            pkg = pkg.replace(os.path.altsep, ".")
         packages.append(pkg)
 
 if 'VERSION' in os.environ:
@@ -29,9 +27,12 @@
     author='Rémy Le Calloch',
     author_email='remy@lecalloch.net',
     license=open('LICENSE.txt').read(),
-    install_requires=[], # dependency management in conda recipe
+    install_requires=[],  # dependency management in conda recipe
     url='https://github.com/Pan6ora/ab-plugin-Notebook',
     long_description=open('README.md').read(),
     description='Launch Jupyter Notebook directly in Activity-Browser',
-    package_data={'': ['jupyter_config/jupyter_notebook_config.py', 'jupyter_config/custom/custom.js']},
-    )
+    package_dir={"": "."},
+    package_data={
+        'ab_plugin_notebook': ['jupyter_config/*',
+                               'jupyter_config/custom/*']},
+)