Fixed sushi generator

I fixed the use of generators =)