Improve code readability in tests