diff --git a/src/builder/build_traits.rs b/src/builder/build_traits.rs index 09ff8c129..d3589fb39 100644 --- a/src/builder/build_traits.rs +++ b/src/builder/build_traits.rs @@ -86,10 +86,13 @@ pub trait Container { name: impl Into, signature: Signature, ) -> Result, BuildError> { - let f_node = self.add_child_op(ops::FuncDefn { - name: name.into(), - signature: signature.clone().into(), - })?; + let f_node = self.add_child_node(NodeType::new( + ops::FuncDefn { + name: name.into(), + signature: signature.signature.clone(), + }, + signature.input_extensions.clone(), + ))?; let db = DFGBuilder::create_with_io( self.hugr_mut(),