Cleaned up project tree

2 files changed, 4 insertions, 4 deletions

diff --git a/tests/test_database.py b/tests/test_database.py
index e33fdae..290b985 100644
--- a/tests/test_database.py
+++ b/tests/test_database.py
@@ -2,7 +2,7 @@ import sqlite3
 
 import pytest
 
-from database.database import Database
+from personal_agent.database.database import Database
 
 
 def test_migrate_creates_schema(tmp_path):
diff --git a/tests/test_memory_store.py b/tests/test_memory_store.py
index f4862cb..8f3407d 100644
--- a/tests/test_memory_store.py
+++ b/tests/test_memory_store.py
@@ -1,9 +1,9 @@
 from datetime import UTC, datetime
 from uuid import UUID, uuid7
 
-from database.database import Database
-from memory import LongTermMemory, MemoryCategory
-from memory_store import LongTermMemoryStore
+from personal_agent.database.database import Database
+from personal_agent.memory.models import LongTermMemory, MemoryCategory
+from personal_agent.memory.store import LongTermMemoryStore
 
 
 def create_store(tmp_path):