Fix missing case in flow control code