Updated generator to take from constants package.