Would be good to add this behavior to the docstring as well.