diff --git a/week4/community-contributions/pytest_generator/pytest_generator.ipynb b/week4/community-contributions/pytest_generator/pytest_generator.ipynb
index 86d1819..7051957 100644
--- a/week4/community-contributions/pytest_generator/pytest_generator.ipynb
+++ b/week4/community-contributions/pytest_generator/pytest_generator.ipynb
@@ -1,5 +1,15 @@
 {
  "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "b8be8252",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!uv pip install pytest"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -59,9 +69,9 @@
     "\n",
     "# Groq\n",
     "_register(\"Groq • Llama 3.1 8B\",   groq_client,   \"llama-3.1-8b-instant\")\n",
-    "_register(\"Groq • Llama 3.1 70B\",  groq_client,   \"llama-3.1-70b-versatile\")\n",
-    "_register(\"Groq • GPT-OSS 20B\",    groq_client,   \"gpt-oss-20b\")\n",
-    "_register(\"Groq • GPT-OSS 120B\",   groq_client,   \"gpt-oss-120b\")\n",
+    "_register(\"Groq • Llama 3.3 70B\",  groq_client,   \"llama-3.3-70b-versatile\")\n",
+    "_register(\"Groq • GPT-OSS 20B\",    groq_client,   \"openai/gpt-oss-20b\")\n",
+    "_register(\"Groq • GPT-OSS 120B\",   groq_client,   \"openai/gpt-oss-120b\")\n",
     "\n",
     "DEFAULT_MODEL = next(iter(MODEL_REGISTRY.keys()), None)\n",
     "\n",