Minor updates in tests.