updated generator test