diff --git a/numba_rvsdg/core/datastructures/ast_transforms.py b/numba_rvsdg/core/datastructures/ast_transforms.py
index 2aa743e..4e4b94e 100644
--- a/numba_rvsdg/core/datastructures/ast_transforms.py
+++ b/numba_rvsdg/core/datastructures/ast_transforms.py
@@ -141,7 +141,7 @@ def prune_empty(self) -> set[WritableASTBlock]:
                 # Empty blocks can only have a single jump target.
                 it = block.jump_targets[0]
                 # Iterate over the blocks looking for blocks that point to the
-                # removed block.
+                # removed block. Then rewire the jump_targets accordingly.
                 for b in list(self.values()):
                     if len(b.jump_targets) == 0:
                         continue
@@ -220,7 +220,7 @@ def transform(self) -> None:
         """Transform Python function stored as self.code."""
         # Convert source code into AST.
         tree = ast.parse(textwrap.dedent(inspect.getsource(self.code))).body
-        # Assert that the code handed in was a function, we can only convert
+        # Assert that the code handed in was a function, we can only transform
         # functions.
         assert isinstance(tree[0], ast.FunctionDef)
         # Run recursive code generation.