diff --git a/pyproject.toml b/pyproject.toml
index bc8d9c7658dcd..5e8a2bfa0e481 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -78,6 +78,7 @@ module = [
     "lightning_app.components.serve.streamlit",
     "lightning_app.components.serve.types.image",
     "lightning_app.components.serve.types.type",
+    "lightning_app.components.serve.python_server",
     "lightning_app.components.training",
     "lightning_app.core.api",
     "lightning_app.core.app",