Code refactoring in tests.