diff --git a/categories_map.py b/categories_map.py
index 0a9fc49afd18..23702082f3d0 100644
--- a/categories_map.py
+++ b/categories_map.py
@@ -849,11 +849,19 @@
     "Validation/Shashlik",
   ],
   "heterogeneous": [
-    "CUDADataFormats/Common",
     "CUDADataFormats/BeamSpot",
+    "CUDADataFormats/CaloCommon",
+    "CUDADataFormats/Common",
+    "CUDADataFormats/EcalDigi",
+    "CUDADataFormats/EcalRecHitSoA",
+    "CUDADataFormats/HcalDigi",
+    "CUDADataFormats/HcalRecHitSoA",
     "CUDADataFormats/SiPixelCluster",
     "CUDADataFormats/SiPixelDigi",
     "CUDADataFormats/StdDictionaries",
+    "CUDADataFormats/Track",
+    "CUDADataFormats/TrackingRecHit",
+    "CUDADataFormats/Vertex",
     "HeterogeneousCore/CUDACore",
     "HeterogeneousCore/CUDAServices",
     "HeterogeneousCore/CUDATest",
@@ -1026,8 +1034,16 @@
   ],
   "reconstruction": [
     "CUDADataFormats/BeamSpot",
+    "CUDADataFormats/CaloCommon",
+    "CUDADataFormats/EcalDigi",
+    "CUDADataFormats/EcalRecHitSoA",
+    "CUDADataFormats/HcalDigi",
+    "CUDADataFormats/HcalRecHitSoA",
     "CUDADataFormats/SiPixelCluster",
     "CUDADataFormats/SiPixelDigi",
+    "CUDADataFormats/Track",
+    "CUDADataFormats/TrackingRecHit",
+    "CUDADataFormats/Vertex",
     "CommonTools/BaseParticlePropagator",
     "CommonTools/CandAlgos",
     "CommonTools/CandUtils",