minor improvement to unit test