fix: code robustness

[EZ4Jam1n]

What type of PR is this?

fix

Check the PR title.

• This PR title match the format: <type>(optional scope): <description>
• The description of this PR title is user-oriented and clear enough for others to understand.
• Attach the PR updating the user documentation if the current PR requires user awareness at the usage level. User docs repo

(Optional) Translate the PR title into Chinese.

(Optional) More detailed description for this PR(en: English/zh: Chinese).

en:
zh(optional):
修复部分代码

(Optional) Which issue(s) this PR fixes:

(optional) The PR that updates user documentation:

[GuangmingLuo]

@EZ4Jam1n check this