2 small fixes in the tests