diff --git a/docker-entrypoint b/docker-entrypoint
index 8e15ce7..85718b6 100755
--- a/docker-entrypoint
+++ b/docker-entrypoint
@@ -1,7 +1,7 @@
 #!/usr/bin/env sh
 set -e
 
-DOCKER_SOCK=/tmp/docker.sock
+DOCKER_SOCK=${DOCKER_SOCK:-/var/run/docker.sock}
 TMP_DIR=/tmp/restart
 
 if [ "$1" = 'autoheal' ] && [ -e ${DOCKER_SOCK} ]; then