Update generator to work with Corona