Fix the docs