Fix ImportError when building Sphinx

I forgot to rename the module in the document.