Fix generator usage example