Fix doc build with temp workaround