Fix num root ports in extract_dfg

CQCL · Oct 22, 2024 · 894f05d · 894f05d
1 parent cf4f32c
commit 894f05d
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tket2/src/circuit/extract_dfg.rs b/tket2/src/circuit/extract_dfg.rs
@@ -24,6 +24,11 @@ pub(super) fn rewrite_into_dfg(circ: &mut Circuit) -> Result<(), CircuitMutError
         _ => signature,
     };
 
+    circ.hugr.set_num_ports(
+        circ.parent(),
+        signature.input_count() + 1,
+        signature.input_count() + 1,
+    );
     circ.hugr.replace_op(circ.parent(), DFG { signature })?;
 
     Ok(())