Better fix

Hide doc things behind with_doc flag