Megre modify code.

Megre modify code.