Fix model counts