Updated generator name in tests