diff --git a/include/vkcv/ShaderStage.hpp b/include/vkcv/ShaderStage.hpp
index ea82fb761b033605a0cbfafbb53a5163704e6358..a52db5d8b0f86f75e89034bc845144f57f149978 100644
--- a/include/vkcv/ShaderStage.hpp
+++ b/include/vkcv/ShaderStage.hpp
@@ -2,13 +2,15 @@
 
 namespace vkcv {
 
-	enum class ShaderStage : uint32_t
+	enum class ShaderStage : VkShaderStageFlags
 	{
-		VERTEX = 0x00000001,
-		TESS_CONTROL = 0x00000002,
-		TESS_EVAL = 0x00000004,
-		GEOMETRY = 0x00000008,
-		FRAGMENT = 0x00000016,
-		COMPUTE = 0x00000032
+		VERTEX = VK_SHADER_STAGE_VERTEX_BIT,
+		TESS_CONTROL = VK_SHADER_STAGE_TESSELLATION_CONTROL_BIT,
+		TESS_EVAL = VK_SHADER_STAGE_TESSELLATION_EVALUATION_BIT,
+		GEOMETRY = VK_SHADER_STAGE_GEOMETRY_BIT,
+		FRAGMENT = VK_SHADER_STAGE_FRAGMENT_BIT,
+		COMPUTE = VK_SHADER_STAGE_COMPUTE_BIT
 	};
+
+	using ShaderStages = vk::Flags<ShaderStage>;
 }