update tests for 0.17.1 generator