Really fix the test. Sorry for the breakage...