diff --git a/labs/lab3/nlp-lab3.ipynb b/labs/lab3/nlp-lab3.ipynb
index adf055085d415efa7ebe4b002374fa26b78f3dd8..186db20755715860f9ef8811088c958b6dafad76 100644
--- a/labs/lab3/nlp-lab3.ipynb
+++ b/labs/lab3/nlp-lab3.ipynb
@@ -76,7 +76,7 @@
     "@dataclass\n",
     "class TrainingConfig:\n",
     "    device: torch.device = torch.device(\"cuda\")\n",
-    "    shard_dir: str = \"data\"\n",
+    "    shard_dir: str = \"/courses/TDDE09/labs/lab3/data\"\n",
     "\n",
     "    # Training steps and data processing\n",
     "    n_steps: int = 4768\n",
@@ -210,7 +210,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "sum(s.numel() for s in shards(\"/courses/TDDE09/labs/lab3/data\"))"
+    "sum(s.numel() for s in shards(TrainingConfig().shard_dir))"
    ]
   },
   {