Fix test (yet again)