-- fix count for new loss