diff --git a/tests/vkrenderframework.cpp b/tests/vkrenderframework.cpp
index 71c5619..5f68054 100644
--- a/tests/vkrenderframework.cpp
+++ b/tests/vkrenderframework.cpp
@@ -1166,6 +1166,10 @@
     m_shaderStages.push_back(shader->GetStageCreateInfo());
 }
 
+void VkPipelineObj::AddShader(VkPipelineShaderStageCreateInfo const & createInfo) {
+    m_shaderStages.push_back(createInfo);
+}
+
 void VkPipelineObj::AddVertexInputAttribs(VkVertexInputAttributeDescription *vi_attrib, uint32_t count) {
     m_vi_state.pVertexAttributeDescriptions = vi_attrib;
     m_vi_state.vertexAttributeDescriptionCount = count;