Modified test to use image io