Fixed up docs generation for all tests