Change assert expected value for total count