Fix error that came in when removing StatelessImage