diff --git a/pyproject.toml b/pyproject.toml
index acb937c0..da74ba14 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "cai-framework"
-version = "0.4.0"
+version = "0.5.0"
 description = "Cybersecurity AI Framework"
 readme = "README.md"
 requires-python = ">=3.9"
diff --git a/src/cai/agents/memory.py b/src/cai/agents/memory.py
index 84c79288..e08e4880 100644
--- a/src/cai/agents/memory.py
+++ b/src/cai/agents/memory.py
@@ -84,7 +84,6 @@ Environment Variables enabling the episodic memory store
 import os
 from cai.sdk.agents import Agent, OpenAIChatCompletionsModel
 from cai.tools.misc.rag import add_to_memory_semantic, add_to_memory_episodic
-from cai.rag.vector_db import get_previous_memory
 
 # Get model from environment or use default
 model = os.getenv('CAI_MODEL', "alias0")