fix tests
small fixes