Fix generated test