doctest fix

Just a simple fix. Doctests were incorrectly displayed.