Fixed a bug with the build image process.