minor change in last code snippet