diff --git a/source/Lib/EncoderLib/SEIEncoder.cpp b/source/Lib/EncoderLib/SEIEncoder.cpp
index bcf1aa3ab92f86743186bd6068cce852819f8c6c..6f92e4ed785a799b38061223e351900609aaf10a 100644
--- a/source/Lib/EncoderLib/SEIEncoder.cpp
+++ b/source/Lib/EncoderLib/SEIEncoder.cpp
@@ -1313,7 +1313,7 @@ void SEIEncoder::initSEINeuralNetworkPostFilterCharacteristics(SEINeuralNetworkP
       sei->m_outSubCFlag = m_pcCfg->getNNPostFilterSEICharacteristicsOutSubCFlag(filterIdx);
     }
 #if JVET_AC0154
-    if((sei->m_purpose & 0x20) != 0)
+    if((sei->m_purpose & NNPC_PurposeType::COLOURIZATION) != 0)
     {
       sei->m_outColourFormatIdc = m_pcCfg->getNNPostFilterSEICharacteristicsOutColourFormatIdc(filterIdx);
     }
diff --git a/source/Lib/EncoderLib/SEIwrite.cpp b/source/Lib/EncoderLib/SEIwrite.cpp
index 743ce2d3c0b89183aec60a960c9262c9162bb109..5be7b918e523fa4620ca2ae873302cdcea417911 100644
--- a/source/Lib/EncoderLib/SEIwrite.cpp
+++ b/source/Lib/EncoderLib/SEIwrite.cpp
@@ -1701,7 +1701,7 @@ void SEIWriter::xWriteSEINeuralNetworkPostFilterCharacteristics(const SEINeuralN
       xWriteFlag(sei.m_outSubCFlag, "nnpfc_out_sub_c_flag");
     }
 #if JVET_AC0154
-    if((sei.m_purpose & 0x20) != 0)
+    if((sei.m_purpose & NNPC_PurposeType::COLOURIZATION) != 0)
     {
       xWriteCode(uint32_t(sei.m_outColourFormatIdc), 2, "nnpfc_out_colour_format_idc");
     }