Minor fix for upward flow