diff --git a/pyproject.toml b/pyproject.toml
index 2e001e2..54f9588 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -20,7 +20,8 @@ dependencies = [
     "fastapi",
     "matplotlib",
     "Pillow",
-    "jax"
+    "jax",
+    "commentjson"
 ]
 
 [tool.setuptools.packages.find]