diff --git a/docs/lecture17.ipynb b/docs/lecture17.ipynb
index ebad9b3..8e14605 100755
--- a/docs/lecture17.ipynb
+++ b/docs/lecture17.ipynb
@@ -128,9 +128,11 @@
     "        cmin=-vmax,\n",
     "        cmax=+vmax,\n",
     "        showscale=colorize,\n",
-    "        colorscale=\"RdBu_r\"\n",
-    "        if colorize\n",
-    "        else [[0, \"rgb(0, 0, 0)\"], [1, DEFAULT_PLOTLY_COLORS[i - 1]]],\n",
+    "        colorscale=(\n",
+    "            \"RdBu_r\"\n",
+    "            if colorize\n",
+    "            else [[0, \"rgb(0, 0, 0)\"], [1, DEFAULT_PLOTLY_COLORS[i - 1]]]\n",
+    "        ),\n",
     "        surfacecolor=t.real if colorize else np.ones(shape=t.shape),\n",
     "    )\n",
     "    S_im = [\n",