WIP : wrap ufuncs correctly