Fix test (forgotten in previous commit)