diff --git a/sup_tutorial/resnet18_cifar10_SE_solution.ipynb b/sup_tutorial/resnet18_cifar10_SE_solution.ipynb
index fdae216..60d7b13 100644
--- a/sup_tutorial/resnet18_cifar10_SE_solution.ipynb
+++ b/sup_tutorial/resnet18_cifar10_SE_solution.ipynb
@@ -371,7 +371,7 @@
       "source": [
         "def check_length(length, value, name):\n",
         "  if len(value) != length:\n",
-        "    raise ValueError(f\"`{name}` must be of length 4 not {len(value)}\")\n",
+        "    raise ValueError(f\"`{name}` must be of length {length} not {len(value)}\")\n",
         "\n",
         "class BlockV2(hk.Module):\n",
         "  \"\"\"ResNet V2 block with optional bottleneck.\"\"\"\n",
@@ -957,4 +957,4 @@
       "outputs": []
     }
   ]
-}
\ No newline at end of file
+}
diff --git a/sup_tutorial/resnet18_cifar10_adversarial.ipynb b/sup_tutorial/resnet18_cifar10_adversarial.ipynb
index f9c9701..cf4b0e4 100644
--- a/sup_tutorial/resnet18_cifar10_adversarial.ipynb
+++ b/sup_tutorial/resnet18_cifar10_adversarial.ipynb
@@ -150,7 +150,7 @@
         "#@title Define the model: Resnet18\n",
         "def check_length(length, value, name):\n",
         "  if len(value) != length:\n",
-        "    raise ValueError(f\"`{name}` must be of length 4 not {len(value)}\")\n",
+        "    raise ValueError(f\"`{name}` must be of length {length} not {len(value)}\")\n",
         "\n",
         "class BlockV2(hk.Module):\n",
         "  \"\"\"ResNet V2 block with optional bottleneck.\"\"\"\n",
@@ -682,4 +682,4 @@
       "outputs": []
     }
   ]
-}
\ No newline at end of file
+}
diff --git a/sup_tutorial/resnet18_cifar10_adversarial_solution.ipynb b/sup_tutorial/resnet18_cifar10_adversarial_solution.ipynb
index d1a14a1..0e0e71e 100644
--- a/sup_tutorial/resnet18_cifar10_adversarial_solution.ipynb
+++ b/sup_tutorial/resnet18_cifar10_adversarial_solution.ipynb
@@ -150,7 +150,7 @@
         "#@title Define the model: Resnet18\n",
         "def check_length(length, value, name):\n",
         "  if len(value) != length:\n",
-        "    raise ValueError(f\"`{name}` must be of length 4 not {len(value)}\")\n",
+        "    raise ValueError(f\"`{name}` must be of length {length} not {len(value)}\")\n",
         "\n",
         "class BlockV2(hk.Module):\n",
         "  \"\"\"ResNet V2 block with optional bottleneck.\"\"\"\n",
@@ -690,4 +690,4 @@
       "outputs": []
     }
   ]
-}
\ No newline at end of file
+}
diff --git a/sup_tutorial/resnet18_cifar10_baseline_solution.ipynb b/sup_tutorial/resnet18_cifar10_baseline_solution.ipynb
index 9989b82..1997251 100644
--- a/sup_tutorial/resnet18_cifar10_baseline_solution.ipynb
+++ b/sup_tutorial/resnet18_cifar10_baseline_solution.ipynb
@@ -334,7 +334,7 @@
       "source": [
         "def check_length(length, value, name):\n",
         "  if len(value) != length:\n",
-        "    raise ValueError(f\"`{name}` must be of length 4 not {len(value)}\")\n",
+        "    raise ValueError(f\"`{name}` must be of length {length} not {len(value)}\")\n",
         "\n",
         "class BlockV2(hk.Module):\n",
         "  \"\"\"ResNet V2 block with optional bottleneck.\"\"\"\n",
@@ -981,4 +981,4 @@
       "outputs": []
     }
   ]
-}
\ No newline at end of file
+}