fixed tests for count update