diff --git a/src/utils/instantiators.py b/src/utils/instantiators.py
index 9fe57510a..adabbf059 100644
--- a/src/utils/instantiators.py
+++ b/src/utils/instantiators.py
@@ -1,9 +1,9 @@
 from typing import List
 
 import hydra
+from lightning import Callback
+from lightning.pytorch.loggers import Logger
 from omegaconf import DictConfig
-from pytorch_lightning import Callback
-from pytorch_lightning.loggers import Logger
 
 from src.utils import pylogger