Update generator tests