diff --git a/src/praisonai/pyproject.toml b/src/praisonai/pyproject.toml
index c3077a4e3..6ce364423 100644
--- a/src/praisonai/pyproject.toml
+++ b/src/praisonai/pyproject.toml
@@ -89,7 +89,7 @@ call = [
     "rich",
     "openai>=1.54.0",
 ]
-train = ["unsloth[colab-new]>=2024.11.7"]
+train = []
 crewai = ["crewai>=0.32.0", "praisonai-tools>=0.0.15"]
 autogen = ["pyautogen>=0.2.19", "praisonai-tools>=0.0.15", "crewai"]
 
@@ -147,7 +147,7 @@ sqlalchemy = {version = ">=2.0.36", optional = true}
 playwright = {version = ">=1.47.0", optional = true}
 openai = {version = ">=1.54.0", optional = true}
 pydantic = {version = "<=2.10.1", optional = true}
-unsloth = {version = ">=2024.11.7", extras = ["colab-new"], optional = true}
+# unsloth = {version = ">=2024.11.7", extras = ["colab-new"], optional = true}
 
 [tool.poetry.group.docs.dependencies]
 mkdocs = "*"
@@ -251,7 +251,7 @@ code = [
     "playwright",
     "pydantic"
 ]
-train = ["unsloth[colab-new]>=2024.11.7"]
+train = []
 realtime = [
     "chainlit", 
     "litellm", 
diff --git a/src/praisonai/tests/unit/test_ollama_fix.py b/src/praisonai/tests/unit/test_ollama_fix.py
index e290d7b07..02ec024f9 100644
--- a/src/praisonai/tests/unit/test_ollama_fix.py
+++ b/src/praisonai/tests/unit/test_ollama_fix.py
@@ -6,7 +6,7 @@
 import os
 
 # Add the source directory to Python path
-sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..', 'src', 'praisonai-agents'))
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..', '..', 'praisonai-agents'))
 
 from praisonaiagents.llm.llm import LLM
 from praisonaiagents.agent.agent import Agent
@@ -69,7 +69,7 @@ def test_tool_call_parsing():
         }
     }
     name, args, id = llm._parse_tool_call_arguments(tool_call_bad, is_ollama=False)
-    assert name == "hello_world", f"Expected 'hello_world', got '{name}'"
+    assert name == "unknown_function", f"Expected 'unknown_function' for malformed JSON, got '{name}'"
     assert args == {}, f"Expected empty dict, got {args}"
     print("✓ Error handling works")