diff --git a/env.yml b/env.yml
index 99656eb..ad80029 100644
--- a/env.yml
+++ b/env.yml
@@ -18,7 +18,7 @@ dependencies:
   - torchvision
   - torchaudio
   - pytorch-cuda=11.8
-  - pytorch-lightning
+  - lightning
   # deep learning libraries
   - transformers
   - datasets