diff --git a/pl_bolts/callbacks/data_monitor.py b/pl_bolts/callbacks/data_monitor.py
index 97ba087229..eab1030216 100644
--- a/pl_bolts/callbacks/data_monitor.py
+++ b/pl_bolts/callbacks/data_monitor.py
@@ -14,6 +14,7 @@
 from pl_bolts.utils.stability import under_review
 from pl_bolts.utils.warnings import warn_missing_pkg
 
+# Backward compatibility for Lightning Logger
 try:
     from pytorch_lightning.loggers import Logger
 except ImportError:
diff --git a/pl_bolts/models/detection/retinanet/retinanet_module.py b/pl_bolts/models/detection/retinanet/retinanet_module.py
index d1711707ba..47c767b2c2 100644
--- a/pl_bolts/models/detection/retinanet/retinanet_module.py
+++ b/pl_bolts/models/detection/retinanet/retinanet_module.py
@@ -135,7 +135,11 @@ def configure_optimizers(self):
 
 @under_review()
 def cli_main():
-    from pytorch_lightning.utilities.cli import LightningCLI
+    # Backward compatibility for Lightning CLI
+    try:
+        from pytorch_lightning.utilities.cli import LightningCLI
+    except ImportError:
+        from pytorch_lightning.cli import LightningCLI
 
     from pl_bolts.datamodules import VOCDetectionDataModule