Fix regression with doc generation