Assign value of yield statement to image (bugfix)