Might be better to use `count_total` here?