fix tests to reflect the new generator name