put test generation back in