diff --git a/setup.py b/setup.py
index f5a1da2d..476f46c2 100644
--- a/setup.py
+++ b/setup.py
@@ -4,7 +4,7 @@
 
 setup(
   name = 'django-webpack-loader',
-  packages = ['webpack_loader', 'webpack_loader/templatetags'], # this must be the same as the name above
+  packages = ['webpack_loader', 'webpack_loader/templatetags', 'webpack_loader/contrib'],
   version = version,
   description = 'Load your webpack bundles and chunks in django',
   author = 'Owais Lone',
diff --git a/webpack_loader/contrib/jinja2ext.py b/webpack_loader/contrib/jinja2ext.py
index adfdbb1f..38a530ca 100644
--- a/webpack_loader/contrib/jinja2ext.py
+++ b/webpack_loader/contrib/jinja2ext.py
@@ -1,4 +1,4 @@
-import jinja2
+import jinja2.ext
 
 from ..templatetags.webpack_loader import render_bundle