Fix docs build