fixing the return image