fixup! Fix issue with documentation