Fix the docs to build