diff --git a/PhysicsAnalysis/D3PDMaker/TruthD3PDMaker/python/MCTruthClassifierConfig.py b/PhysicsAnalysis/D3PDMaker/TruthD3PDMaker/python/MCTruthClassifierConfig.py
index 60c57aa9036ee59e63c3faac82c4305f850dba02..4e0e51e2329c66a5f0100198683952a3821c0593 100644
--- a/PhysicsAnalysis/D3PDMaker/TruthD3PDMaker/python/MCTruthClassifierConfig.py
+++ b/PhysicsAnalysis/D3PDMaker/TruthD3PDMaker/python/MCTruthClassifierConfig.py
@@ -1,4 +1,4 @@
-# Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
+# Copyright (C) 2002-2019 CERN for the benefit of the ATLAS collaboration
 
 # $Id$
 #
@@ -19,8 +19,10 @@ mckey = resolveSGKey ('DataVector<xAOD::TruthParticle_v1>', D3PDMakerFlags.Truth
 
 
 from TruthD3PDAnalysis import D3PDMCTruthClassifier
+from MCTruthClassifier import MCTruthClassifierBase
 D3PDMCTruthClassifier = D3PDMCTruthClassifier (name = 'D3PDMCTruthClassifier',
                                                xAODTruthParticleContainerName = mckey,
+                                               ParticleCaloExtensionTool=MCTruthClassifierBase.ClassifierParticleCaloExtensionTool,
                                                pTNeutralPartCut = 1e-3,
                                                OutputLevel = ERROR,
                                                partExtrConePhi = 0.6, #0.4