1 files changed, 4 insertions, 2 deletions
diff --git a/src/rendering/vertex_buffer.h b/src/rendering/vertex_buffer.h
index 77a76c9..39f43ba 100644
--- a/src/rendering/vertex_buffer.h
+++ b/src/rendering/vertex_buffer.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& get_layout()      const = 0;
         virtual void set_layout(const VertexBufferLayout& layout) = 0;
 
-        static VertexBuffer* create(const void* data, uint32_t size);
+        static Ref<VertexBuffer> create(const void* data, uint32_t size);
     };
 };