minor change based on review