Fix issues with adding tally tests