Fix for images not working correctly