Small fixes in tests.