diff --git a/salty/base/python_env/init.sls b/salty/base/python_env/init.sls
index 4532916..7dce343 100644
--- a/salty/base/python_env/init.sls
+++ b/salty/base/python_env/init.sls
@@ -1,3 +1,5 @@
+{% from "python_env/map.jinja" import python_env with context %}
+
 pip-install:
   pkg.installed:
     - name: {{ python_env.pip }}