minor improvement in test