Fixing json call in count_matrix test