fix test to check cluster sizes