Fix a bug cause by code struct change.