[FIX] fix issue #61 -7

[15100399015]

What is the purpose of the change

Related issues/PRs

Related issues: #590
Related pr:#591

Brief change log

Checklist

• I have read the Contributing Guidelines on pull requests.
• I have explained the need for this PR and the problem it solves
• I have explained the changes or the new features added to this PR
• I have added tests corresponding to this change
• I have updated the documentation to reflect this change
• I have verified that this change is backward compatible
• If this is a code change: I have written unit tests to fully verify the new behavior.

[dlimeng]

LGTM