minor change of code