Fix the docs too