修正代碼。