diff --git a/docs/source-pytorch/cli/lightning_cli_intermediate_2.rst b/docs/source-pytorch/cli/lightning_cli_intermediate_2.rst
index c60c339607..b61baa5eb5 100644
--- a/docs/source-pytorch/cli/lightning_cli_intermediate_2.rst
+++ b/docs/source-pytorch/cli/lightning_cli_intermediate_2.rst
@@ -57,7 +57,7 @@ To support multiple models, when instantiating ``LightningCLI`` omit the ``model
 
     # main.py
     from pytorch_lightning.cli import LightningCLI
-    from pytorch_lightning.demos.boring_classes import DemoModel
+    from pytorch_lightning.demos.boring_classes import DemoModel, BoringDataModule
 
 
     class Model1(DemoModel):
@@ -101,7 +101,7 @@ To support multiple data modules, when instantiating ``LightningCLI`` omit the `
     # main.py
     import torch
     from pytorch_lightning.cli import LightningCLI
-    from pytorch_lightning.demos.boring_classes import BoringDataModule
+    from pytorch_lightning.demos.boring_classes import DemoModel, BoringDataModule
 
 
     class FakeDataset1(BoringDataModule):