diff --git a/Reconstruction/VKalVrt/GNNVertexConstructor/CMakeLists.txt b/Reconstruction/VKalVrt/GNNVertexConstructor/CMakeLists.txt
index 5924c4c1757dda0211ace613857d4ce156025b49..ea15740690d2f2a3d51278d29e479fa55e41b171 100644
--- a/Reconstruction/VKalVrt/GNNVertexConstructor/CMakeLists.txt
+++ b/Reconstruction/VKalVrt/GNNVertexConstructor/CMakeLists.txt
@@ -91,6 +91,7 @@ atlas_add_library(   GNNVertexConstructorLib
 # Install files from the package:
 atlas_install_python_modules( python/*.py POST_BUILD_CMD ${ATLAS_FLAKE8} )
 
+atlas_install_data( data/* )
 
 
 
diff --git a/Reconstruction/VKalVrt/GNNVertexConstructor/Root/GNNVertexConstructorTool.cxx b/Reconstruction/VKalVrt/GNNVertexConstructor/Root/GNNVertexConstructorTool.cxx
index 8a64626f53e6f829de97a446092b9d6019737f8a..b9bb2c7de31cd766962933e6e4aa73cd869fd301 100644
--- a/Reconstruction/VKalVrt/GNNVertexConstructor/Root/GNNVertexConstructorTool.cxx
+++ b/Reconstruction/VKalVrt/GNNVertexConstructor/Root/GNNVertexConstructorTool.cxx
@@ -23,7 +23,7 @@ GNNVertexConstructorTool::GNNVertexConstructorTool(const std::string &type, cons
 
   declareProperty("JetTrackLinks", m_trackLinksKey = "BTagging_AntiKt4EMPFlowAuxDyn.TrackLinks");
   declareProperty("JetVertexLinks",
-                  m_vertexLinksKey = "BTagging_AntiKt4EMPFlowAuxDyn.track_vertexing");
+                  m_vertexLinksKey = "BTagging_AntiKt4EMPFlowAuxDyn.vertex_indices");
 
   declareProperty("GNNTool", m_gnn_Tool, "GNN Tool");
   declareProperty("VertexFitterTool", m_vertexFitterTool, "Vertex fitting tool");
diff --git a/Reconstruction/VKalVrt/GNNVertexConstructor/data/network.onnx b/Reconstruction/VKalVrt/GNNVertexConstructor/data/network.onnx
new file mode 100644
index 0000000000000000000000000000000000000000..b2d8e9b5f3f287347a59a62d2ebf8824cc45cb69
Binary files /dev/null and b/Reconstruction/VKalVrt/GNNVertexConstructor/data/network.onnx differ
diff --git a/Reconstruction/VKalVrt/GNNVertexConstructor/python/GNNVertexConstructorToolConfig.py b/Reconstruction/VKalVrt/GNNVertexConstructor/python/GNNVertexConstructorToolConfig.py
index 2eb1ef8258152a5eb7d16fc48d6bb19120273f99..fa5a18bcc22cda0f7466175e3453a400d82937c1 100644
--- a/Reconstruction/VKalVrt/GNNVertexConstructor/python/GNNVertexConstructorToolConfig.py
+++ b/Reconstruction/VKalVrt/GNNVertexConstructor/python/GNNVertexConstructorToolConfig.py
@@ -12,7 +12,7 @@ def GNNVertexConstructorToolCfg(flags, name="LMEdevTool", **kwargs):
     gnnTool = acc.getPrimaryAndMerge(
             GNNToolCfg(
                 flags,
-                NNFile           ="../network.onnx",
+                NNFile           ="GNNVertexConstructor/network.onnx",
                 trackLinkType    ="IPARTICLE",  #Either IPARTICLE or  TRACK_PARTICLE
                 variableRemapping={"BTagTrackToJetAssociator" : "GhostTrack"},
                 )