This MR just fixes two bugs where the pT of the jet is used instead of the energy.
Tagging @mswiatlo @sschramm