Fix accidental failed unit tests from #60