Update wrapper to output at right size.