fix generator tests