Fix crash in Image Flow