修正一些 code