modify a little

modify a little