Fixing generator test