update code

bug fixed