Small improvements in tests
2 files changed