diff --git a/PhysicsTools/PatAlgos/python/producersHeavyIons/heavyIonJets_cff.py b/PhysicsTools/PatAlgos/python/producersHeavyIons/heavyIonJets_cff.py
index a2b5fc30305d0..05680c7e97ef5 100644
--- a/PhysicsTools/PatAlgos/python/producersHeavyIons/heavyIonJets_cff.py
+++ b/PhysicsTools/PatAlgos/python/producersHeavyIons/heavyIonJets_cff.py
@@ -38,11 +38,6 @@
 from RecoBTag.ImpactParameter.trackCounting3D3rdComputer_cfi import *
 from PhysicsTools.PatAlgos.recoLayer0.jetCorrFactors_cfi import *
 
-ak5JetTracksAssociatorAtVertex.jets = "akCs4PFJets"
-ak5JetTracksAssociatorAtVertex.tracks = "highPurityGeneralTracks"
-
-impactParameterTagInfos.jetTracks = "ak5JetTracksAssociatorAtVertex"
-
 recoPFJetsHIpostAODTask = cms.Task(
     PFTowers,
     pfEmptyCollection,
diff --git a/RecoBTag/ImpactParameter/python/impactParameterTagInfos_cfi.py b/RecoBTag/ImpactParameter/python/impactParameterTagInfos_cfi.py
index 597725bad1b6d..9e9a1f8c90151 100644
--- a/RecoBTag/ImpactParameter/python/impactParameterTagInfos_cfi.py
+++ b/RecoBTag/ImpactParameter/python/impactParameterTagInfos_cfi.py
@@ -16,3 +16,7 @@
     jetDirectionUsingGhostTrack = cms.bool(False),
     useTrackQuality = cms.bool(False)
 )
+
+from Configuration.Eras.Modifier_pp_on_AA_2018_cff import pp_on_AA_2018
+from Configuration.Eras.Modifier_pp_on_PbPb_run3_cff import pp_on_PbPb_run3
+(pp_on_AA_2018 | pp_on_PbPb_run3).toModify(impactParameterTagInfos, jetTracks = "ak5JetTracksAssociatorAtVertex")
diff --git a/RecoJets/JetAssociationProducers/python/ak5JTA_cff.py b/RecoJets/JetAssociationProducers/python/ak5JTA_cff.py
index 75d19cbb364b2..29228c9866218 100644
--- a/RecoJets/JetAssociationProducers/python/ak5JTA_cff.py
+++ b/RecoJets/JetAssociationProducers/python/ak5JTA_cff.py
@@ -40,3 +40,10 @@
 
 ak5JTAExplicitTask = cms.Task(ak5JetTracksAssociatorExplicit)
 ak5JTAExplicit = cms.Sequence(ak5JTAExplicitTask)
+
+from Configuration.Eras.Modifier_pp_on_AA_2018_cff import pp_on_AA_2018
+from Configuration.Eras.Modifier_pp_on_PbPb_run3_cff import pp_on_PbPb_run3
+(pp_on_AA_2018 | pp_on_PbPb_run3).toModify(ak5JetTracksAssociatorAtVertex,
+                                           jets = "akCs4PFJets",
+                                           tracks = "highPurityGeneralTracks"
+                                       )