diff --git a/frontend/src/core/cells/__tests__/runs.test.ts b/frontend/src/core/cells/__tests__/runs.test.ts
index f0e9f70757a..e5145d5e9fd 100644
--- a/frontend/src/core/cells/__tests__/runs.test.ts
+++ b/frontend/src/core/cells/__tests__/runs.test.ts
@@ -15,7 +15,7 @@ const { reducer, initialState, isPureMarkdown } = exportedForTesting;
 
 function first<T>(map: Map<string, T> | undefined): T {
   invariant(map, "Map is undefined");
-  return map.values().next().value;
+  return map.values().next().value as T;
 }
 
 describe("RunsState Reducer", () => {
diff --git a/tests/_runtime/snapshots/docstrings_class.txt b/tests/_runtime/snapshots/docstrings_class.txt
index fce203d2687..b3138a86aa4 100644
--- a/tests/_runtime/snapshots/docstrings_class.txt
+++ b/tests/_runtime/snapshots/docstrings_class.txt
@@ -1,4 +1,4 @@
-<div class="codehilite"><pre><span></span><code><span class="k">class</span> <span class="nc">MyClass</span><span class="p">()</span>
+<div class="codehilite"><pre><span></span><code><span class="k">class</span><span class="w"> </span><span class="nc">MyClass</span><span class="p">()</span>
 </code></pre></div>
 
 <span class="paragraph">Some docstring for the class.</span>
diff --git a/tests/_runtime/snapshots/docstrings_function.txt b/tests/_runtime/snapshots/docstrings_function.txt
index cfc19f32421..f7d462a2e8e 100644
--- a/tests/_runtime/snapshots/docstrings_function.txt
+++ b/tests/_runtime/snapshots/docstrings_function.txt
@@ -1,4 +1,4 @@
-<div class="codehilite"><pre><span></span><code><span class="k">def</span> <span class="nf">my_func</span><span class="p">(</span><span class="n">arg1</span><span class="p">,</span> <span class="n">arg2</span><span class="p">)</span>
+<div class="codehilite"><pre><span></span><code><span class="k">def</span><span class="w"> </span><span class="nf">my_func</span><span class="p">(</span><span class="n">arg1</span><span class="p">,</span> <span class="n">arg2</span><span class="p">)</span>
 </code></pre></div>
 
 <span class="paragraph">This is a simple docstring for a function.</span>
\ No newline at end of file
diff --git a/tests/_runtime/snapshots/docstrings_function_external.txt b/tests/_runtime/snapshots/docstrings_function_external.txt
index 4c191ca40d8..f97673b1d31 100644
--- a/tests/_runtime/snapshots/docstrings_function_external.txt
+++ b/tests/_runtime/snapshots/docstrings_function_external.txt
@@ -1,4 +1,4 @@
-<div class="codehilite"><pre><span></span><code><span class="k">def</span> <span class="nf">my_func</span><span class="p">(</span><span class="n">arg1</span><span class="p">,</span> <span class="n">arg2</span><span class="p">)</span>
+<div class="codehilite"><pre><span></span><code><span class="k">def</span><span class="w"> </span><span class="nf">my_func</span><span class="p">(</span><span class="n">arg1</span><span class="p">,</span> <span class="n">arg2</span><span class="p">)</span>
 </code></pre></div>
 
 <div class='external-docs'><div class="document">
diff --git a/tests/_runtime/snapshots/docstrings_function_google.txt b/tests/_runtime/snapshots/docstrings_function_google.txt
index 326801eeceb..bde9313829a 100644
--- a/tests/_runtime/snapshots/docstrings_function_google.txt
+++ b/tests/_runtime/snapshots/docstrings_function_google.txt
@@ -1,4 +1,4 @@
-<div class="codehilite"><pre><span></span><code><span class="k">def</span> <span class="nf">my_func</span><span class="p">(</span><span class="n">arg1</span><span class="p">,</span> <span class="n">arg2</span><span class="p">)</span>
+<div class="codehilite"><pre><span></span><code><span class="k">def</span><span class="w"> </span><span class="nf">my_func</span><span class="p">(</span><span class="n">arg1</span><span class="p">,</span> <span class="n">arg2</span><span class="p">)</span>
 </code></pre></div>
 
 <h1 id="arguments">Arguments</h1>