wip: bugfixes to ufunc support