[MINOR] minor update code