From 5b8e9bd2d89f39304b2da24ecf4008a5be25ddb2 Mon Sep 17 00:00:00 2001
From: Sasha Bylinkin <37345380+abylinkin@users.noreply.github.com>
Date: Mon, 14 Oct 2024 12:47:47 +0200
Subject: [PATCH] Update DGCandProducer.cxx

---
 PWGUD/TableProducer/DGCandProducer.cxx | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/PWGUD/TableProducer/DGCandProducer.cxx b/PWGUD/TableProducer/DGCandProducer.cxx
index beb9b84aff7..34b3a3318b9 100644
--- a/PWGUD/TableProducer/DGCandProducer.cxx
+++ b/PWGUD/TableProducer/DGCandProducer.cxx
@@ -14,6 +14,7 @@
 
 #include "Framework/runDataProcessing.h"
 #include "Framework/AnalysisTask.h"
+#include "ReconstructionDataFormats/Vertex.h"
 #include "PWGUD/DataModel/UDTables.h"
 #include "PWGUD/Core/UPCHelpers.h"
 #include "PWGUD/Core/DGSelector.h"
@@ -271,8 +272,7 @@ struct DGCandProducer {
       auto rtrwTOF = udhelpers::rPVtrwTOF<true>(tracks, collision.numContrib());
       int upc_flag = 0;
       ushort flags = collision.flags();
-      const ushort UPCModeMask = 0x1 << 1;
-      if (flags & UPCModeMask)
+      if (flags & Vertex::UPCMode)
         upc_flag = 1;
       outputCollisions(bc.globalBC(), bc.runNumber(),
                        collision.posX(), collision.posY(), collision.posZ(), upc_flag,