More correct pT computation