diff --git a/pyproject.toml b/pyproject.toml
index 2f3d951..4acb3ee 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -3,7 +3,7 @@ name = "aether"
 version = '1.0'
 requires-python = ">=3.12"
 dependencies = [
-    "lightning>=2.6.0",
+    "lightning==2.6.1",
     "torch>=2.9.1",
     "hydra-core==1.3.2",
     "hydra-colorlog==1.2.0",