Update unit tests with new generator config