Oops, fix broken unit test