fix tests

(should have been part of b0a4a4b)