diff --git a/VulkanHppGenerator.cpp b/VulkanHppGenerator.cpp
index 435afc7..41033fd 100644
--- a/VulkanHppGenerator.cpp
+++ b/VulkanHppGenerator.cpp
@@ -72,7 +72,7 @@ const std::string flagsHeader(
 "    }\n"
 "\n"
 "    Flags(BitType bit)\n"
-"      : m_mask(static_cast<uint32_t>(bit))\n"
+"      : m_mask(static_cast<MaskType>(bit))\n"
 "    {\n"
 "    }\n"
 "\n"
diff --git a/vulkan/vulkan.hpp b/vulkan/vulkan.hpp
index 5f1e403..ff8d425 100644
--- a/vulkan/vulkan.hpp
+++ b/vulkan/vulkan.hpp
@@ -77,7 +77,7 @@ namespace vk
     }
 
     Flags(BitType bit)
-      : m_mask(static_cast<uint32_t>(bit))
+      : m_mask(static_cast<MaskType>(bit))
     {
     }