Fix generators tests