diff --git a/lab1/PT_Part1_Intro.ipynb b/lab1/PT_Part1_Intro.ipynb
index 49bcdeee..575f5a21 100644
--- a/lab1/PT_Part1_Intro.ipynb
+++ b/lab1/PT_Part1_Intro.ipynb
@@ -425,8 +425,8 @@
         "x_input = torch.tensor([[1, 2.]])\n",
         "model_output = model(x_input)\n",
         "print(f\"input shape: {x_input.shape}\")\n",
-        "print(f\"output shape: {y.shape}\")\n",
-        "print(f\"output result: {y}\")"
+        "print(f\"output shape: {model_output.shape}\")\n",
+        "print(f\"output result: {model_output}\")"
       ]
     },
     {
@@ -547,7 +547,9 @@
         "out_with_identity = # TODO\n",
         "\n",
         "print(f\"input: {x_input}\")\n",
-        "print(\"Network linear output: {}; network identity output: {}\".format(out_with_linear, out_with_identity))"
+        "print(\"Network linear output: {}; network identity output: {}\".format(out_with_linear, out_with_identity))",
+        "\n"
+        "assert torch.equal(x_input, out_with_identity)"
       ]
     },
     {