a little modification.