fix: partition info get count flags usage

Follow up to change [1] fix statements related to
FFA_PARTITION_COUNT_FLAG_MASK and FFA_PARTITION_COUNT_FLAG
macros usage.

[1] https://review.trustedfirmware.org/c/hafnium/hafnium/+/13033

Change-Id: I2a08ea66fc7de8ad9d6eb9cbeb70c1a087e04289
Signed-off-by: Olivier Deprez <olivier.deprez@arm.com>
diff --git a/inc/vmapi/hf/ffa.h b/inc/vmapi/hf/ffa.h
index 8d6d59c..98656a8 100644
--- a/inc/vmapi/hf/ffa.h
+++ b/inc/vmapi/hf/ffa.h
@@ -131,12 +131,12 @@
 #define FFA_MEM_PERM_RX UINT32_C(0x3)
 
 /*
- * Defined in Table 13.31 in the FF-A v1.1 BETA0 specification.
+ * Defined in Table 13.34 in the FF-A v1.1 EAC0 specification.
  * The Partition count flag is used by FFA_PARTITION_INFO_GET to specify
  * if partition info descriptors should be returned or just the count.
  */
-#define FFA_PARTITION_COUNT_FLAG 0x1
-#define FFA_PARTITION_COUNT_FLAG_MASK 0x1
+#define FFA_PARTITION_COUNT_FLAG UINT32_C(0x1)
+#define FFA_PARTITION_COUNT_FLAG_MASK (UINT32_C(0x1) << 0)
 
 /**
  * For use where the FF-A specification refers explicitly to '4K pages'. Not to