Support 'when' syntax after for statement.