Fixing doc generation