diff --git a/classification.py b/classification.py
index 9ba2907..c923f07 100644
--- a/classification.py
+++ b/classification.py
@@ -96,7 +96,7 @@ def _format_results(checkpoint, results):
 def main():
     args = build_parser().parse_args()
     import torch
-    from lightning.pytorch import seed_everything
+    from pytorch_lightning import seed_everything
     from utils.training import (
         build_trainer,
         create_run_paths,
diff --git a/pretrain.py b/pretrain.py
index 372459d..0460c29 100644
--- a/pretrain.py
+++ b/pretrain.py
@@ -99,7 +99,7 @@ def main():
     args = build_parser().parse_args()
     _validate_sampling_args(args)
     import torch
-    from lightning.pytorch import seed_everything
+    from pytorch_lightning import seed_everything
     from utils.training import (
         build_trainer,
         create_run_paths,
diff --git a/segmentation.py b/segmentation.py
index 3ec0d96..32078aa 100644
--- a/segmentation.py
+++ b/segmentation.py
@@ -111,7 +111,7 @@ def _format_results(root_checkpoint, results_list):
 def main():
     args = build_parser().parse_args()
     import torch
-    from lightning.pytorch import seed_everything
+    from pytorch_lightning import seed_everything
     from utils.training import (
         build_trainer,
         create_run_paths,