diff --git a/src/ShaderGen/2d/Std2DMaterial.cpp b/src/ShaderGen/2d/Std2DMaterial.cpp
index 64d93ece..115f24b2 100644
--- a/src/ShaderGen/2d/Std2DMaterial.cpp
+++ b/src/ShaderGen/2d/Std2DMaterial.cpp
@@ -35,7 +35,7 @@ bool Std2DMaterial::CustomVertexShader(ShaderCreateInfoVertex *vsc)
 
     if(cfg->coordinate_system==CoordinateSystem2D::Ortho)
     {
-        mci->AddUBO(VK_SHADER_STAGE_VERTEX_BIT,
+        mci->AddUBO(VK_SHADER_STAGE_ALL_GRAPHICS,
                     DescriptorSetType::Global,
                     SBS_ViewportInfo);
     }
diff --git a/src/ShaderGen/3d/Std3DMaterial.cpp b/src/ShaderGen/3d/Std3DMaterial.cpp
index 6a0542b8..53bd591a 100644
--- a/src/ShaderGen/3d/Std3DMaterial.cpp
+++ b/src/ShaderGen/3d/Std3DMaterial.cpp
@@ -37,7 +37,7 @@ bool Std3DMaterial::CustomVertexShader(ShaderCreateInfoVertex *vsc)
     //    vsc->AddFunction(func::GetNormal);
     //}
 
-    mci->AddUBO(VK_SHADER_STAGE_VERTEX_BIT,
+    mci->AddUBO(VK_SHADER_STAGE_ALL_GRAPHICS,
                 DescriptorSetType::Global,
                 SBS_ViewportInfo);