Do a little changes

Do a little changes