Fix linter error in tests