minor change in code