diff --git a/lab1/PT_Part1_Intro.ipynb b/lab1/PT_Part1_Intro.ipynb
index 49bcdeee..b56b9976 100644
--- a/lab1/PT_Part1_Intro.ipynb
+++ b/lab1/PT_Part1_Intro.ipynb
@@ -423,7 +423,7 @@
       "source": [
         "# Test the model with example input\n",
         "x_input = torch.tensor([[1, 2.]])\n",
-        "model_output = model(x_input)\n",
+        "y = model(x_input)\n",
         "print(f\"input shape: {x_input.shape}\")\n",
         "print(f\"output shape: {y.shape}\")\n",
         "print(f\"output result: {y}\")"
diff --git a/lab1/solutions/PT_Part1_Intro_Solution.ipynb b/lab1/solutions/PT_Part1_Intro_Solution.ipynb
index a8feddf7..2b7704f5 100644
--- a/lab1/solutions/PT_Part1_Intro_Solution.ipynb
+++ b/lab1/solutions/PT_Part1_Intro_Solution.ipynb
@@ -491,7 +491,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 13,
+      "execution_count": null,
       "metadata": {
         "id": "zKhp6XqCFFa0"
       },
@@ -509,7 +509,7 @@
       "source": [
         "# Test the model with example input\n",
         "x_input = torch.tensor([[1, 2.]])\n",
-        "model_output = model(x_input)\n",
+        "y = model(x_input)\n",
         "print(f\"input shape: {x_input.shape}\")\n",
         "print(f\"output shape: {y.shape}\")\n",
         "print(f\"output result: {y}\")"