diff --git a/Core/src/main/java/me/neurodock/plugin/loader/Loader.java b/Core/src/main/java/me/neurodock/plugin/loader/Loader.java
index 6124c55..e5c5eb1 100644
--- a/Core/src/main/java/me/neurodock/plugin/loader/Loader.java
+++ b/Core/src/main/java/me/neurodock/plugin/loader/Loader.java
@@ -40,9 +40,9 @@ public class Loader {
      * Mock example!
      * @return
      */
-    public OllamaTool[] getTools(Plugin plugin) {
+    public OllamaFunctionTool[] getTools(Plugin plugin) {
 
-        ArrayList<OllamaTool> tools = new ArrayList<>();
+        ArrayList<OllamaFunctionTool> tools = new ArrayList<>();
 
         for(Tool tool : plugin.getTools()) {
             tools.add(new OllamaFunctionTool() {
@@ -89,7 +89,7 @@ public class Loader {
             });
         }
 
-        return tools.toArray(tools.toArray(new OllamaTool[0]));
+        return tools.toArray(tools.toArray(new OllamaFunctionTool[0]));
 
     }