Fixing bug in docstring's code example