diff --git a/src/lightning_app/components/training.py b/src/lightning_app/components/training.py
index b467615cca..78790f10a3 100644
--- a/src/lightning_app/components/training.py
+++ b/src/lightning_app/components/training.py
@@ -97,7 +97,7 @@ class PyTorchLightningScriptRunner(TracerPythonScript):
         if self.node_rank != 0:
             return {}, args, kwargs
 
-        from lightning.pytorch.serve import ServableModuleValidator
+        from pytorch_lightning.serve import ServableModuleValidator
 
         callbacks = kwargs.get("callbacks", [])
         if self.sanity_serving: