minor change to code