diff --git a/states/top.sls b/states/top.sls
index 3adcde1..302b1f7 100644
--- a/states/top.sls
+++ b/states/top.sls
@@ -1,6 +1,10 @@
 
 {%- set states = salt['cp.list_states'](saltenv) -%}
-{%- set envs = ['prod','dev'] -%}
+{%- if pillar['branches'] is defined -%}
+{%- if pillar['branches']['saltmaster'] is defined -%}
+{%- set envs = pillar['branches']['saltmaster'] -%}
+{%- endif -%}
+{%- endif -%}
 {% for env in envs %}
 {{ env }}:
   'os_family:RedHat':