diff --git a/python/podio_gen/cpp_generator.py b/python/podio_gen/cpp_generator.py
index 9eb1ca478..4d5531a3b 100644
--- a/python/podio_gen/cpp_generator.py
+++ b/python/podio_gen/cpp_generator.py
@@ -525,8 +525,10 @@ def _write_list(name, target_folder, files, comment):
 
     def _write_all_collections_header(self):
         """Write a header file that includes all collection headers"""
-
-        collection_files = (x.split("::")[-1] + "Collection.h" for x in self.datamodel.datatypes)
+        collection_files = (
+            x.split("::")[-1] + "Collection.h"
+            for x in list(self.datamodel.datatypes.keys()) + list(self.datamodel.links.keys())
+        )
         self._write_file(
             os.path.join(self.install_dir, self.package_name, f"{self.package_name}.h"),
             self._eval_template(