diff --git a/tests/run_tests.sh b/tests/run_tests.sh
index 96cb9ad..f626112 100755
--- a/tests/run_tests.sh
+++ b/tests/run_tests.sh
@@ -56,6 +56,9 @@
     - source: ${template_path}
     - context:
       _data: {{ pillar.get("_data", {}) }}
+    {%- if pillar.get('service_name') %}
+      service_name: {{ pillar.service_name }}
+    {%- endif %}
 EOF
 }