Update count in test