Fix documentation refactoring bug