diff --git a/traefik/config.sls b/traefik/config.sls
index 337b069..6b042a3 100644
--- a/traefik/config.sls
+++ b/traefik/config.sls
@@ -5,6 +5,8 @@ traefik-config:
     - name: {{ traefik.config_file }}
     - source: salt://{{ slspath }}/files/traefik.toml
     - template: jinja
+    - context:
+        config: {{ trafik.config }}
     - user: {{ traefik.user}}
     - group: {{ traefik.group }}
     - mode: 0640
diff --git a/traefik/files/traefik.toml b/traefik/files/traefik.toml
index 048683f..98ab59f 100644
--- a/traefik/files/traefik.toml
+++ b/traefik/files/traefik.toml
@@ -1,7 +1,3 @@
-{%- from "traefik/map.jinja" import traefik with context -%}
-
-{%- set config = salt['pillar.get']('traefik:config', {}) -%}
-
 {% for section in config.values() -%}
 [{{ section }}]
 {% for key, val in section.iteritems() -%}