diff --git a/models.py b/models.py
index c891533..b67a14f 100644
--- a/models.py
+++ b/models.py
@@ -2,6 +2,8 @@ from flask_sqlalchemy import SQLAlchemy
 
 import tiktoken
 
+import json
+
 db = SQLAlchemy()
 
 class Room(db.Model):