Fix broken doctests
Merge pull request #1175 from frewsxcv/patch-2

Fix broken sphinx docs, thanks @frewsxcv