Whoops, I forgot to fix this unit test...