Cleaning up generator step