Almost. I think this needs to be inside `countDecorator`