1 files changed, 4 insertions, 2 deletions
diff --git a/src/Rendering/VertexBuffer.h b/src/Rendering/VertexBuffer.h
index 35b9a84..db85755 100644
--- a/src/Rendering/VertexBuffer.h
+++ b/src/Rendering/VertexBuffer.h
@@ -1,5 +1,7 @@
 #pragma once
 
+#include "Core/Memory.h"
+
 #include <cstdint>
 #include <vector>
 
@@ -22,7 +24,7 @@ namespace Donut
         ~VertexBufferLayout() = default;
 
         template<typename T>
-        void Push(uint32_t count) 
+        void Push(uint32_t count)
         {
             static_assert(false);
         }
@@ -68,6 +70,6 @@ namespace Donut
         virtual const VertexBufferLayout& GetLayout()      const = 0;
         virtual void SetLayout(const VertexBufferLayout& layout) = 0;
 
-        static VertexBuffer* Create(const void* data, uint32_t size);
+        static Ref<VertexBuffer> Create(const void* data, uint32_t size);
     };
 };