diff --git a/scripts/mbedtls_dev/macro_collector.py b/scripts/mbedtls_dev/macro_collector.py
index ca277795c..b98e40e5f 100644
--- a/scripts/mbedtls_dev/macro_collector.py
+++ b/scripts/mbedtls_dev/macro_collector.py
@@ -50,7 +50,7 @@ class PSAMacroCollector:
                 return True
             if '_CATEGORY_' in name:
                 return True
-        return name.endswith('_FLAG') or name.endswith('MASK')
+        return name.endswith('_FLAG') or name.endswith('_MASK')
 
     # "#define" followed by a macro name with either no parameters
     # or a single parameter and a non-empty expansion.