diff --git a/image-builder/image_builder.sh b/image-builder/image_builder.sh
index c5d0ceb2f4..d484d40ccc 100755
--- a/image-builder/image_builder.sh
+++ b/image-builder/image_builder.sh
@@ -113,7 +113,7 @@ if [ -n "${USE_DOCKER}" ] ; then
 fi
 # The kata rootfs image expect init and kata-agent to be installed
 init="${ROOTFS}/sbin/init"
-[ -x "${init}" ] || [ -L ${init} ] || die "/sbin/init is not installed in ${ROOTFS_DIR}"
+[ -x "${init}" ] || [ -L ${init} ] || die "/sbin/init is not installed in ${ROOTFS}"
 OK "init is installed"
 [ "${AGENT_INIT}" == "yes" ] || [ -x "${ROOTFS}/usr/bin/${AGENT_BIN}" ] || \
 	die "/usr/bin/${AGENT_BIN} is not installed in ${ROOTFS}