diff --git a/docker-service-with-dist.gradle b/docker-service-with-dist.gradle
index 4beb9ac2..c06bd4bd 100644
--- a/docker-service-with-dist.gradle
+++ b/docker-service-with-dist.gradle
@@ -25,6 +25,9 @@ ADD index-construction-process.tar /dist
 
 RUN mkdir /wmsa
 
+# This will make the service grab the hostname from the HOSTNAME variable
+ENV WMSA_IN_DOCKER true
+
 ENTRYPOINT WMSA_HOME=/wmsa /${application.applicationName}/bin/${application.applicationName} \${arg0} \${arg1}
 """
     }