fix wrapped return type