Update counts to use directive