fix tests

Found with Eastwood.