Use a better module for this generator