update tests to changes in generator in f42fa96e243cdfa4b42a25df79e41f4e59fcd327