Fix test generation.