diff --git a/opencl_fdtd/simulation.py b/opencl_fdtd/simulation.py
index 8023d81..096ce7d 100644
--- a/opencl_fdtd/simulation.py
+++ b/opencl_fdtd/simulation.py
@@ -19,7 +19,7 @@ __author__ = 'Jan Petykiewicz'
 
 
 # Create jinja2 env on module load
-jinja_env = jinja2.Environment(loader=jinja2.PackageLoader(__name__, 'kernels'))
+jinja_env = jinja2.Environment(loader=jinja2.PackageLoader(__name__.split('.')[0], 'kernels'))
 
 
 class Simulation(object):