modify some code

modify some code