diff --git a/examples/VulkanTest/main.cpp b/examples/VulkanTest/main.cpp
index 8690fc699..e731f04ed 100644
--- a/examples/VulkanTest/main.cpp
+++ b/examples/VulkanTest/main.cpp
@@ -142,7 +142,7 @@ int main()
 	const Nz::VertexBuffer* drfreakVB = drfreakMesh->GetVertexBuffer();
 	const Nz::IndexBuffer* drfreakIB = drfreakMesh->GetIndexBuffer();
 
-	// Vertex buffer
+	// Index buffer
 	std::cout << "Index count: " << drfreakIB->GetIndexCount() << std::endl;
 
 	Nz::RenderBuffer* renderBufferIB = static_cast<Nz::RenderBuffer*>(drfreakIB->GetBuffer()->GetImpl());
@@ -154,7 +154,7 @@ int main()
 
 	Nz::VulkanBuffer* indexBufferImpl = static_cast<Nz::VulkanBuffer*>(renderBufferIB->GetHardwareBuffer(&vulkanDevice));
 
-	// Index buffer
+	// Vertex buffer
 	std::cout << "Vertex count: " << drfreakVB->GetVertexCount() << std::endl;
 
 	Nz::RenderBuffer* renderBufferVB = static_cast<Nz::RenderBuffer*>(drfreakVB->GetBuffer()->GetImpl());
diff --git a/include/Nazara/Renderer/RenderStates.hpp b/include/Nazara/Renderer/RenderStates.hpp
index d31b37be0..f4d4ea369 100644
--- a/include/Nazara/Renderer/RenderStates.hpp
+++ b/include/Nazara/Renderer/RenderStates.hpp
@@ -9,9 +9,13 @@
 
 #include <Nazara/Prerequisites.hpp>
 #include <Nazara/Utility/Enums.hpp>
+#include <memory>
+#include <vector>
 
 namespace Nz
 {
+	class ShaderStageImpl;
+
 	struct RenderStates
 	{
 		BlendFunc dstBlend = BlendFunc_Zero;
@@ -32,6 +36,8 @@ namespace Nz
 			UInt32 writeMask = 0xFFFFFFFF;
 		} stencilBack, stencilFront;
 
+		std::vector<std::shared_ptr<ShaderStageImpl>> shaderStages;
+
 		bool blending    = false;
 		bool colorWrite  = true;
 		bool depthBuffer = false;