diff --git a/build_library/base_image_util.sh b/build_library/base_image_util.sh
index febeb154e5..93c13ea43a 100755
--- a/build_library/base_image_util.sh
+++ b/build_library/base_image_util.sh
@@ -210,7 +210,7 @@ create_base_image() {
 
   # Record directories installed to the stateful partition.
   # Ignore /var/tmp, systemd covers this entry.
-  sudo "${SCRIPTS_DIR}/gen_tmpfiles.py" --root="${root_fs_dir}" \
+  sudo "${BUILD_LIBRARY_DIR}/gen_tmpfiles.py" --root="${root_fs_dir}" \
       --output="${root_fs_dir}/usr/lib/tmpfiles.d/base_image.conf" \
       --ignore=/var/tmp "${root_fs_dir}/var"
 
diff --git a/gen_tmpfiles.py b/build_library/gen_tmpfiles.py
similarity index 100%
rename from gen_tmpfiles.py
rename to build_library/gen_tmpfiles.py