diff --git a/Reconstruction/VKalVrt/GNNVertexConstructor/Root/GNNVertexConstructorTool.cxx b/Reconstruction/VKalVrt/GNNVertexConstructor/Root/GNNVertexConstructorTool.cxx
index cd34e5747fc9c44af1792a3ddf6a2aff8fcfedfe..73d5e3af58d2e567e8666742e4a523955d013694 100644
--- a/Reconstruction/VKalVrt/GNNVertexConstructor/Root/GNNVertexConstructorTool.cxx
+++ b/Reconstruction/VKalVrt/GNNVertexConstructor/Root/GNNVertexConstructorTool.cxx
@@ -27,7 +27,7 @@ GNNVertexConstructorTool::GNNVertexConstructorTool(const std::string &type, cons
       m_minD0(0.1)
       {
   declareInterface<IGNNVertexConstructorInterface>(this);
-  declareProperty("JetTrackLinks", m_trackLinksKey = "AntiKt4EMPFlowJetsAuxDyn.GN2v01_TrackOrigin");
+  declareProperty("JetTrackLinks", m_trackLinksKey = "AntiKt4EMPFlowJetsAuxDyn.TrackLinks");
   declareProperty("JetVertexLinks",m_vertexLinksKey = "AntiKt4EMPFlowJetsAuxDyn.GN2v01_VertexIndex");
   declareProperty("GNNTool", m_gnn_Tool, "GNN Tool");
   declareProperty("VertexFitterTool", m_vertexFitterTool, "Vertex fitting tool");
@@ -159,7 +159,7 @@ StatusCode GNNVertexConstructorTool::performVertexFit(const xAOD::JetContainer *
 
   //Read Decor Handle for Track links and Vertex links
   SG::ReadDecorHandle<xAOD::JetContainer, TLC> trackLinksHandle(m_trackLinksKey, ctx);
-  SG::ReadDecorHandle<xAOD::JetContainer, std::vector<char, std::allocator<char>>>
+  SG::ReadDecorHandle<xAOD::JetContainer, std::vector<char>>
       vertexLinksHandle(m_vertexLinksKey, ctx);
   SG::WriteDecorHandle< xAOD::JetContainer, std::vector<ElementLink<xAOD::VertexContainer>>> 
       jetWriteDecorHandleVertexLink (m_jetWriteDecorKeyVertexLink, ctx);
diff --git a/Reconstruction/VKalVrt/GNNVertexConstructor/src/GNNVertexConstructorAlg.cxx b/Reconstruction/VKalVrt/GNNVertexConstructor/src/GNNVertexConstructorAlg.cxx
index 8ea6f71b38101ee4237f836d84a6b49042c26909..f1a9186f77a11baaf15b238c0dfa62693ff06f5a 100644
--- a/Reconstruction/VKalVrt/GNNVertexConstructor/src/GNNVertexConstructorAlg.cxx
+++ b/Reconstruction/VKalVrt/GNNVertexConstructor/src/GNNVertexConstructorAlg.cxx
@@ -59,7 +59,7 @@ StatusCode GNNVertexConstructorAlg::execute(const EventContext &ctx) const {
 
   //Decorates the Jets using the GNN model
   //May b removed in future with development of GNN model
-  ATH_CHECK(m_VtxTool->decorateJets(inJetContainer.ptr()));
+  //ATH_CHECK(m_VtxTool->decorateJets(inJetContainer.ptr()));
   
   //Perform a Vertex fit
   ATH_CHECK(m_VtxTool->performVertexFit(inJetContainer.ptr(), outVertexContainer.ptr(), *pv,  ctx));