a minor change in the code