diff --git a/traefik/files/traefik.toml b/traefik/files/traefik.toml
index e230faa..dddd172 100644
--- a/traefik/files/traefik.toml
+++ b/traefik/files/traefik.toml
@@ -1,4 +1,7 @@
-{% set config = pillar.get('traefik.config', {}) -%}
+{%- from "traefik/map.jinja" import traefik with context -%}
+
+{%- set config = salt['pillar.get']('traefik:config', {}) -%}
+
 {% for section in config.iteritems() -%}
 [{{ section }}]
 {% for key, val in section.iteritems() -%}