Fix assumption about vertex count calculation