diff --git a/Gems/Atom/RHI/DX12/Code/Source/RHI/Conversions.cpp b/Gems/Atom/RHI/DX12/Code/Source/RHI/Conversions.cpp
index 9776734158..a29c7ae402 100644
--- a/Gems/Atom/RHI/DX12/Code/Source/RHI/Conversions.cpp
+++ b/Gems/Atom/RHI/DX12/Code/Source/RHI/Conversions.cpp
@@ -1357,9 +1357,9 @@ namespace AZ
             return table[(uint32_t)mask];
         }
     
-        uint32_t ConvertColorWriteMask(uint8_t writeMask)
+        uint8_t ConvertColorWriteMask(uint8_t writeMask)
         {            
-            uint32_t dflags = 0;
+            uint8_t dflags = 0;
             if(writeMask == 0)
             {
                 return dflags;
diff --git a/Gems/Atom/RHI/DX12/Code/Source/RHI/Conversions.h b/Gems/Atom/RHI/DX12/Code/Source/RHI/Conversions.h
index 887598de79..640a5fef31 100644
--- a/Gems/Atom/RHI/DX12/Code/Source/RHI/Conversions.h
+++ b/Gems/Atom/RHI/DX12/Code/Source/RHI/Conversions.h
@@ -165,6 +165,6 @@ namespace AZ
             D3D12_SHADER_VISIBILITY shaderVisibility,
             D3D12_STATIC_SAMPLER_DESC& staticSamplerDesc);
     
-        uint32_t ConvertColorWriteMask(uint8_t writeMask);
+        uint8_t ConvertColorWriteMask(uint8_t writeMask);
     }
 }