a little modification

a little modification