1 files changed, 8 insertions, 8 deletions

diff --git a/src/rendering/shader.cpp b/src/rendering/shader.cpp
index 1e84125..1bbd0fc 100644
--- a/src/rendering/shader.cpp
+++ b/src/rendering/shader.cpp
@@ -1,8 +1,8 @@
 #include "shader.h"
 #include "renderer.h"
+
 #include "platform/opengl/opengl_shader.h"
 #include "platform/vulkan/vulkan_shader.h"
-#include <memory>
 
 namespace Donut
 {
@@ -45,32 +45,32 @@ namespace Donut
         }
     }
 
-    void ShaderLibrary::add(const std::shared_ptr<Shader>& shader)
+    void ShaderLibrary::add(const Ref<Shader>& shader)
     {
         auto& name = shader->get_name();
         add(name, shader);
     }
 
-    void ShaderLibrary::add(const std::string& name, const std::shared_ptr<Shader>& shader)
+    void ShaderLibrary::add(const std::string& name, const Ref<Shader>& shader)
     {
         m_shaders[name] = shader;
     }
 
-    std::shared_ptr<Shader> ShaderLibrary::load(const std::string& filepath)
+    Ref<Shader> ShaderLibrary::load(const std::string& filepath)
     {
-        auto shader = std::shared_ptr<Shader>(Shader::create(filepath));
+        auto shader = Ref<Shader>(Shader::create(filepath));
         add(shader);
         return shader;
     }
 
-    std::shared_ptr<Shader> ShaderLibrary::load(const std::string& name, const std::string& filepath)
+    Ref<Shader> ShaderLibrary::load(const std::string& name, const std::string& filepath)
     {
-        auto shader = std::shared_ptr<Shader>(Shader::create(filepath));
+        auto shader = Ref<Shader>(Shader::create(filepath));
         add(name, shader);
         return shader;
     }
 
-    std::shared_ptr<Shader> ShaderLibrary::get(const std::string& name)
+    Ref<Shader> ShaderLibrary::get(const std::string& name)
     {
         if (exists(name))
             return m_shaders[name];