add special check for B class