small fixes in tests