Generator version looks to work the same.
simple variable change