generator test fix