Small fix on flow