Fix docstring for wrapped functions