diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index b0792c99..1f819cd8 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -9,7 +9,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        python-version: ["3.9", "3.10", "3.11"]
+        python-version: ["3.9", "3.10", "3.11", "3.12"]
 
     steps:
       - uses: actions/checkout@v4
diff --git a/setup.py b/setup.py
index 58bfc75b..a9aed26b 100644
--- a/setup.py
+++ b/setup.py
@@ -14,7 +14,7 @@
     author_email="martin.glauer@ovgu.de",
     description="",
     zip_safe=False,
-    python_requires=">=3.9, <3.12",
+    python_requires=">=3.9, <3.13",
     install_requires=[
         "certifi",
         "idna",
@@ -40,7 +40,7 @@
         "matplotlib",
         "rdkit",
         "selfies",
-        "lightning<=2.1",
+        "lightning<=2.1,>=2.5",
         "jsonargparse[signatures]>=4.17",
         "omegaconf",
         "seaborn",