diff --git a/database.py b/database.py
index 13e79d9..50c1220 100644
--- a/database.py
+++ b/database.py
@@ -1,5 +1,4 @@
 import sqlite3
-import json
 import os
 from typing import Optional, List, Tuple
 from datetime import datetime
@@ -7,7 +6,7 @@ import numpy as np
 from openai import OpenAI
 
 # Database configuration
-DB_PATH = os.getenv("CHAT_DB_PATH", "chat_history.db")
+DB_PATH = os.getenv("DB_PATH", "chat_history.db")
 EMBEDDING_MODEL = os.getenv("EMBEDDING_MODEL", "qwen3-embed-4b")
 EMBEDDING_DIMENSION = 2048  # Default for qwen3-embed-4b
 MAX_HISTORY_MESSAGES = int(os.getenv("MAX_HISTORY_MESSAGES", "1000"))
@@ -234,11 +233,13 @@ class ChatDatabase:
         cursor = conn.cursor()
 
         # Join chat_messages and message_embeddings to get content and embeddings
-        cursor.execute("""
+        cursor.execute(
+            """
             SELECT cm.message_id, cm.content, me.embedding 
             FROM chat_messages cm
             JOIN message_embeddings me ON cm.message_id = me.message_id
-        """)
+        """
+        )
         rows = cursor.fetchall()
 
         results = []
diff --git a/vibe-bot.container b/vibe-bot.container
index 19eddb5..c88599c 100644
--- a/vibe-bot.container
+++ b/vibe-bot.container
@@ -4,8 +4,10 @@ After=network.target
 
 [Container]
 Image=localhost/vibe-bot:latest
+Volume=vibe-bot-db:/db
 Environment=DISCORD_TOKEN=your_token_here
 Environment=OPENAI_API_ENDPOINT=http://host.containers.internal:8000
+Environment=DB_PATH=/db/chat_history.db
 
 [Service]
 Restart=on-failure