Make docstring assignment slightly more logical