Add helper method to get generator.