Fix BuildImage test