Wrap write_only inner fn

Internal function wrapping fix for write_only decorator.