Small improvement in test.