Whoops, fix a unit test