Fixes test for generate.