a little optimize of the last commit.

Signed-off-by: Ray Allen <ray@ray-Lenovo-IdeaPad-S400.(none)>