Small fix for flow input