Change jet collection and improve jet variable definition.