Minor code improvement.

Signed-off-by: Nigel <Nigel@10.203.2.25>
Code optimization

Signed-off-by: Nigel <Nigel@10.203.2.25>