Update for doc_builder changes