diff --git a/tools/release.py b/tools/release.py
index 4f2a23d..cfb75ae 100644
--- a/tools/release.py
+++ b/tools/release.py
@@ -3,7 +3,7 @@
 import sys
 from datetime import date
 
-import toml
+import tomlkit as toml
 import changelog
 from semver import Version
 
@@ -64,7 +64,7 @@ def run_command(command: str):
     f.write(str(version))
 
 with open("pyproject.toml", "r") as f:
-    pyproject = toml.load(f)
+    pyproject = toml.load(f).unwrap()
 pyproject["tool"]["poetry"]["version"] = str(version)
 
 with open("pyproject.toml", "w") as f: