Add Tally test and fix error
Trivial change