Small wrapper change