update test for moved generator
v0.6.0