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;
diff --git a/tests/vkrenderframework.h b/tests/vkrenderframework.h
index 3fcff30..4455c04 100644
--- a/tests/vkrenderframework.h
+++ b/tests/vkrenderframework.h
@@ -400,6 +400,7 @@
    public:
     VkPipelineObj(VkDeviceObj *device);
     void AddShader(VkShaderObj *shaderObj);
+    void AddShader(VkPipelineShaderStageCreateInfo const & createInfo);
     void AddVertexInputAttribs(VkVertexInputAttributeDescription *vi_attrib, uint32_t count);
     void AddVertexInputBindings(VkVertexInputBindingDescription *vi_binding, uint32_t count);
     void AddColorAttachment(uint32_t binding, const VkPipelineColorBlendAttachmentState *att);