v154: Bug 14417 - Added VkShaderStage to VkShaderCreateInfo
diff --git a/tests/init.cpp b/tests/init.cpp
index 21c50d9..b6f9294 100644
--- a/tests/init.cpp
+++ b/tests/init.cpp
@@ -73,7 +73,7 @@
     void CreateCommandBufferTest();
     void CreatePipelineTest();
     void CreateShaderTest();
-    void CreateShader(VkShader *pshader);
+    void CreateShader(VkShader *pshader, VkShaderStage stage);
 
     VkDevice device() {return m_device->handle();}
 
@@ -528,7 +528,7 @@
     CreateCommandBufferTest();
 }
 
-void VkTest::CreateShader(VkShader *pshader)
+void VkTest::CreateShader(VkShader *pshader, VkShaderStage stage)
 {
     void *code;
     uint32_t codeSize;
@@ -564,6 +564,7 @@
     createInfo.module = module;
     createInfo.pName = "main";
     createInfo.flags = 0;
+    createInfo.stage = stage;
     err = vkCreateShader(device(), &createInfo, &shader);
     ASSERT_VK_SUCCESS(err);