Fix doc generation issue