minor change, add test