Fix docs generation