Fix docstring and transform