small fix on a function wrapper