diff --git a/.kitchen.yml b/.kitchen.yml
index c115e42..aec9de8 100644
--- a/.kitchen.yml
+++ b/.kitchen.yml
@@ -18,6 +18,7 @@
     - name: heka
       repo: git
       source: https://gerrit.mcp.mirantis.com/salt-formulas/heka
+      branch: <%=ENV['GERRIT_BRANCH'] || 'master' %>
   state_top:
     base:
       "*":