Fix error with documentation