Fixes doc generation