diff --git a/tests/vktestframeworkandroid.cpp b/tests/vktestframeworkandroid.cpp
index fabfa95..8b09d79 100644
--- a/tests/vktestframeworkandroid.cpp
+++ b/tests/vktestframeworkandroid.cpp
@@ -28,9 +28,9 @@
 
 bool VkTestFramework::m_use_glsl = true;
 
-bool VkTestFramework::GLSLtoSPV(const VkShaderStage shader_type,
-                                 const char *pshader,
-                                 std::vector<unsigned int> &spirv)
+bool VkTestFramework::GLSLtoSPV(const VkShaderStageFlagBits shader_type,
+                                const char *pshader,
+                                std::vector<unsigned int> &spv)
 {
     assert(false);
 }
diff --git a/tests/vktestframeworkandroid.h b/tests/vktestframeworkandroid.h
index bcd3f03..00ade20 100644
--- a/tests/vktestframeworkandroid.h
+++ b/tests/vktestframeworkandroid.h
@@ -47,7 +47,7 @@
     static void InitArgs(int *argc, char *argv[]);
     static void Finish();
 
-    bool GLSLtoSPV(const VkShaderStage shader_type,
+    bool GLSLtoSPV(const VkShaderStageFlagBits shader_type,
                    const char *pshader,
                    std::vector<unsigned int> &spv);