diff --git a/examples/pl_bug_report/bug_report_model.py b/examples/pl_bug_report/bug_report_model.py
index 7739630237..551ea21721 100644
--- a/examples/pl_bug_report/bug_report_model.py
+++ b/examples/pl_bug_report/bug_report_model.py
@@ -3,7 +3,7 @@ import os
 import torch
 from torch.utils.data import DataLoader, Dataset
 
-from pytorch_lightning import LightningModule, Trainer
+from lightning.pytorch import LightningModule, Trainer
 
 
 class RandomDataset(Dataset):