fix generator test