modify  some code

modify some code