Add handling for count capacity check on preview