Adjust test for generators