Add "return" in ImageService