Fixed bug where setting the images didn't work