Fix test-driver (see commit `Generation: add tests.`).