Fix test after 60b203d1b7c8fdad543578b85590f22b26e90ba5