Remove duplication of setting the max source count