diff --git a/docs/source-app/code_samples/convert_pl_to_app/train.py b/docs/source-app/code_samples/convert_pl_to_app/train.py
index e9f9de3a6a..ea4a983c9d 100644
--- a/docs/source-app/code_samples/convert_pl_to_app/train.py
+++ b/docs/source-app/code_samples/convert_pl_to_app/train.py
@@ -29,7 +29,7 @@ class LitAutoEncoder(pl.LightningModule):
         x = x.view(x.size(0), -1)
         z = self.encoder(x)
         x_hat = self.decoder(z)
-        loss = F.mse_loss(x_hat, x)
+        loss = F.mse_loss(x_hat, y)
         self.log("train_loss", loss)
         return loss