Small changes to test doc generation