Fix return type error for function def node