diff --git a/lazyDir/centos-7.Dockerfile b/lazyDir/centos-7.Dockerfile
index 80e3b72..f1efaa2 100644
--- a/lazyDir/centos-7.Dockerfile
+++ b/lazyDir/centos-7.Dockerfile
@@ -12,6 +12,7 @@
 	git \
 	wget \
 	unzip \
+	which \
 	&& yum -q clean packages
 
 # Add user to build and package
@@ -21,3 +22,6 @@
 ARG group=dummy
 
 RUN groupadd -g "${gid}" "${group}" && useradd -ms /bin/bash -g "${group}" -u "${uid}" "${user}"
+
+# Get script directory from lazyLib
+ARG dir=.
diff --git a/lazyDir/ubuntu-16.Dockerfile b/lazyDir/ubuntu-16.Dockerfile
index ed7b9c1..bdb260a 100644
--- a/lazyDir/ubuntu-16.Dockerfile
+++ b/lazyDir/ubuntu-16.Dockerfile
@@ -21,3 +21,6 @@
 ARG group=dummy
 
 RUN groupadd -g "${gid}" "${group}" && useradd -ms /bin/bash -g "${group}" -u "${uid}" "${user}"
+
+# Get script directory from lazyLib
+ARG dir=.