Fix doc generation