Fix doc generation bug