4 files changed, 12 insertions, 10 deletions
diff --git a/src/Platform/Vulkan/VulkanRendererAPI.cpp b/src/Platform/Vulkan/VulkanRendererAPI.cpp
index 377f100..86d9653 100644
--- a/src/Platform/Vulkan/VulkanRendererAPI.cpp
+++ b/src/Platform/Vulkan/VulkanRendererAPI.cpp
@@ -32,7 +32,7 @@ namespace Donut
 		// TODO(Hachem): Implement Vulkan face culling setting
 	}
 
-	void VulkanRendererAPI::DrawIndexed(const std::shared_ptr<VertexArray>& vertexArray, uint32_t indexCount)
+	void VulkanRendererAPI::DrawIndexed(const Ref<VertexArray>& vertexArray, uint32_t indexCount)
 	{
 		// TODO(Hachem): Implement Vulkan indexed drawing
 	}
diff --git a/src/Platform/Vulkan/VulkanRendererAPI.h b/src/Platform/Vulkan/VulkanRendererAPI.h
index 73cfbdb..47342ac 100644
--- a/src/Platform/Vulkan/VulkanRendererAPI.h
+++ b/src/Platform/Vulkan/VulkanRendererAPI.h
@@ -1,5 +1,6 @@
 #pragma once
 
+#include "Core/Memory.h"
 #include "Rendering/Renderer.h"
 
 namespace Donut
@@ -16,7 +17,7 @@ namespace Donut
 		virtual void EnableDepthTest()                            override;
 		virtual void SetFaceCulling(bool enabled)                 override;
 
-		virtual void DrawIndexed(const std::shared_ptr<VertexArray>& vertexArray, 
+		virtual void DrawIndexed(const Ref<VertexArray>& vertexArray, 
 								 uint32_t indexCount = 0) override;
 	};
 };
diff --git a/src/Platform/Vulkan/VulkanVertexArray.cpp b/src/Platform/Vulkan/VulkanVertexArray.cpp
index 9fda0a3..de22f05 100644
--- a/src/Platform/Vulkan/VulkanVertexArray.cpp
+++ b/src/Platform/Vulkan/VulkanVertexArray.cpp
@@ -22,13 +22,13 @@ namespace Donut
 		// TODO(Hachem): Implement Vulkan vertex array unbinding
 	}
 
-	void VulkanVertexArray::AddVertexBuffer(const std::shared_ptr<VertexBuffer>& vertexBuffer)
+	void VulkanVertexArray::AddVertexBuffer(const Ref<VertexBuffer>& vertexBuffer)
 	{
 		// TODO(Hachem): Implement Vulkan vertex buffer addition
 		m_VertexBuffers.push_back(vertexBuffer);
 	}
 
-	void VulkanVertexArray::SetIndexBuffer(const std::shared_ptr<IndexBuffer>& indexBuffer)
+	void VulkanVertexArray::SetIndexBuffer(const Ref<IndexBuffer>& indexBuffer)
 	{
 		// TODO(Hachem): Implement Vulkan index buffer setting
 		m_IndexBuffer = indexBuffer;
diff --git a/src/Platform/Vulkan/VulkanVertexArray.h b/src/Platform/Vulkan/VulkanVertexArray.h
index a6eef6a..e4484e5 100644
--- a/src/Platform/Vulkan/VulkanVertexArray.h
+++ b/src/Platform/Vulkan/VulkanVertexArray.h
@@ -1,6 +1,7 @@
 #pragma once
 
 #include "Rendering/VertexArray.h"
+#include "Core/Memory.h"
 
 namespace Donut
 {
@@ -14,14 +15,14 @@ namespace Donut
 		virtual void Bind()   const override;
 		virtual void Unbind() const override;
 
-		virtual void AddVertexBuffer(const std::shared_ptr<VertexBuffer>& vertexBuffer) override;
-		virtual void SetIndexBuffer(const std::shared_ptr<IndexBuffer>& indexBuffer)   override;
+		virtual void AddVertexBuffer(const Ref<VertexBuffer>& vertexBuffer) override;
+		virtual void SetIndexBuffer(const Ref<IndexBuffer>& indexBuffer)   override;
 
-		virtual const std::vector<std::shared_ptr<VertexBuffer>>& GetVertexBuffers() const { return m_VertexBuffers; }
-		virtual const std::shared_ptr<IndexBuffer>& GetIndexBuffer()                 const { return m_IndexBuffer;   }
+		virtual const std::vector<Ref<VertexBuffer>>& GetVertexBuffers() const { return m_VertexBuffers; }
+		virtual const Ref<IndexBuffer>& GetIndexBuffer()                 const { return m_IndexBuffer;   }
 	private:
 		uint32_t                                   m_RendererID;
-		std::vector<std::shared_ptr<VertexBuffer>> m_VertexBuffers;
-		std::shared_ptr<IndexBuffer>               m_IndexBuffer;
+		std::vector<Ref<VertexBuffer>> m_VertexBuffers;
+		Ref<IndexBuffer>               m_IndexBuffer;
 	};
 };