Fix test case for previous commit. Oops.