Small fix on s2. To avoid sets with only one class