diff --git a/src/builder/conditional.rs b/src/builder/conditional.rs
index 607dda8bb..017447a82 100644
--- a/src/builder/conditional.rs
+++ b/src/builder/conditional.rs
@@ -129,7 +129,6 @@ impl<B: AsMut<Hugr> + AsRef<Hugr>> ConditionalBuilder<B> {
                 // TODO: Allow this to be non-pure
                 self.hugr_mut().add_op_before(sibling_node, case_op)?
             } else {
-                // TODO: Allow this to be non-pure
                 self.add_child_op(case_op)?
             };
 
diff --git a/src/builder/dataflow.rs b/src/builder/dataflow.rs
index c6a3124c0..6dfd13853 100644
--- a/src/builder/dataflow.rs
+++ b/src/builder/dataflow.rs
@@ -253,8 +253,7 @@ pub(crate) mod test {
 
                 let inner_builder = func_builder.dfg_builder(
                     FunctionType::new(type_row![NAT], type_row![NAT]),
-                    // TODO: This should be None
-                    Some(ExtensionSet::new()),
+                    None,
                     [int],
                 )?;
                 let inner_id = n_identity(inner_builder)?;