Oops, fix unit tests