Fixing a bug created after fixing PU jet ID