make a little modification